diff --git a/src/gemseo/core/derivatives/jacobian_assembly.py b/src/gemseo/core/derivatives/jacobian_assembly.py
index b5141f236c1047cd5194d460cfa3d25f9ef970bb..1f3c0208817ff7e3a3d61cd52a1a03269720b565 100644
--- a/src/gemseo/core/derivatives/jacobian_assembly.py
+++ b/src/gemseo/core/derivatives/jacobian_assembly.py
@@ -724,7 +724,7 @@ class JacobianAssembly:
         (dfun_dx, dfun_dy) = ({}, {})
         for fun in functions:
             dfun_dx[fun] = self.assemble_jacobian([fun], variables)
-            dfun_dy[fun] = self.assemble_jacobian([fun], couplings_and_res)
+            dfun_dy[fun] = self.assemble_jacobian([fun], couplings_and_states)
 
         mode = self._get_derivation_mode(mode, n_variables, n_functions)