diff --git a/dowhy/gcm/ml/autogluon.py b/dowhy/gcm/ml/autogluon.py
index 3a0f0ef2b1..919f62737a 100644
--- a/dowhy/gcm/ml/autogluon.py
+++ b/dowhy/gcm/ml/autogluon.py
@@ -57,9 +57,9 @@ def fit(self, X: np.ndarray, Y: np.ndarray) -> None:
         )
 
         if self._model_persistence_max_memory is not None:
-            self._auto_gluon_model.persist_models(max_memory=self._model_persistence_max_memory)
+            self._auto_gluon_model.persist(max_memory=self._model_persistence_max_memory)
         else:
-            self._auto_gluon_model.persist_models()
+            self._auto_gluon_model.persist()
 
     def predict(self, X: np.ndarray) -> np.ndarray:
         X = shape_into_2d(X)