diff --git a/scikit_mol/safeinference.py b/scikit_mol/safeinference.py
index 9ac2380..c9188a3 100644
--- a/scikit_mol/safeinference.py
+++ b/scikit_mol/safeinference.py
@@ -74,7 +74,7 @@ def wrapper(obj, X, y=None, *args, **kwargs):
                 # TODO, how can we check y in the same way as the estimator?
                 y = check_array(
                     y,
-                    force_all_finite=False,  # accept_sparse="csr",
+                    ensure_all_finite=False,  # accept_sparse="csr",
                     ensure_2d=False,
                     dtype=None,
                     input_name="y",