diff --git a/tests/test_aakr.py b/tests/test_aakr.py
index 4371465..2fb7e3e 100644
--- a/tests/test_aakr.py
+++ b/tests/test_aakr.py
@@ -4,7 +4,11 @@
 import pytest
 
 from sklearn.datasets import load_linnerud
-from sklearn.utils.testing import assert_allclose
+
+try:  # scikit-learn < 0.24.0
+    from sklearn.utils.testing import assert_allclose
+except ModuleNotFoundError:  # scikit-learn >= 0.24.0
+    from sklearn.utils._testing import assert_allclose
 
 from aakr import AAKR