get diffusion pk as arg when deleting a stream

diff --git a/nonstop/views.py b/nonstop/views.py
index f6da185b21bef0b6be324e1947c24199b2b90163..14a8af34219eea57906838c2db077673ae77380c 100644 (file)
--- a/nonstop/views.py
+++ b/nonstop/views.py
@@ -415,8 +415,8 @@ class AddStreamedDiffusionView(CreateView):
 
 
 class DelStreamedDiffusionView(RedirectView):
-    def get_redirect_url(self):
-        diffusion = Diffusion.objects.get(id=self.kwargs['pk'])
+    def get_redirect_url(self, pk):
+        diffusion = Diffusion.objects.get(id=pk)
         streamed_diffusion = diffusion.streameddiffusion_set.first()
         utils.remove_streamed_diffusion(streamed_diffusion)
         streamed_diffusion.delete()