diff --git a/src/qiboml/operations/differentiation.py b/src/qiboml/operations/differentiation.py
index 17e3e64..3f680b4 100644
--- a/src/qiboml/operations/differentiation.py
+++ b/src/qiboml/operations/differentiation.py
@@ -51,7 +51,7 @@ def evaluate(self, x: ndarray, encoding, training, decoding, backend, *parameter
         circuit = encoding(x) + training
 
         # compute first gradient part, wrt data
-        gradient = self.gradient_wrt_data(
+        gradient = self.gradient_from_data(
             data=x,
             encoding=encoding,
             circuit=circuit,