diff --git a/molfeat/utils/datatype.py b/molfeat/utils/datatype.py
index 11fec76..80a7563 100644
--- a/molfeat/utils/datatype.py
+++ b/molfeat/utils/datatype.py
@@ -281,6 +281,6 @@ def as_numpy_array_if_possible(arr, dtype: Optional[None]):
                 and np.isscalar(arr[0][0])
             ):
                 return sk_utils.check_array(
-                    arr, accept_sparse=True, force_all_finite=False, ensure_2d=False, allow_nd=True
+                    arr, accept_sparse=True, ensure_all_finite=False, ensure_2d=False, allow_nd=True
                 )
     return arr