diff --git a/src/simsopt/geo/curve.py b/src/simsopt/geo/curve.py
index c5d20587a..e23e682a6 100644
--- a/src/simsopt/geo/curve.py
+++ b/src/simsopt/geo/curve.py
@@ -1519,8 +1519,8 @@ def dkappa_by_dcoeff_vjp(self, v):
 
         """
         return Derivative({
-            self: self.dkappa_by_dcoeff_vjp_jax(self.get_dofs(), v),
-            self.surf: self.dkappa_by_dsurf_vjp_jax(self.surf.get_dofs(), v)
+            self: self.dkappa_by_dcoeff_vjp_jax(self.get_dofs(), self.surf.get_dofs(), v),
+            self.surf: self.dkappa_by_dsurf_vjp_jax(self.get_dofs(), self.surf.get_dofs(), v)
             })
 
     # TORSION