diff --git a/tests/preprocessing/test_imputation.py b/tests/preprocessing/test_imputation.py
index d9382084..476b1ae0 100644
--- a/tests/preprocessing/test_imputation.py
+++ b/tests/preprocessing/test_imputation.py
@@ -307,7 +307,7 @@ def test_miceforest_impute_numerical_data(impute_iris_adata):
     "array_type,expected_error",
     [
         (np.array, None),
-        (da.array, None),
+        (da.array, NotImplementedError),
         (sparse.csr_matrix, NotImplementedError),
     ],
 )