diff --git a/glicid-tainted/packages/staging.scm b/glicid-tainted/packages/staging.scm
index df1a67c..3e75b54 100644
--- a/glicid-tainted/packages/staging.scm
+++ b/glicid-tainted/packages/staging.scm
@@ -111,6 +111,10 @@
  (package
   (inherit python-seisbench)
   (name "python-seisbench+cuda")
+  (inputs (modify-inputs (package-inputs python-seisbench)
+              (delete python-pytorch)
+              (prepend python-pytorch-with-cuda11)))
+
  )
 )