diff --git a/python/cuml/test/test_base.py b/python/cuml/test/test_base.py
index 57b452acca..e706ae8bc4 100644
--- a/python/cuml/test/test_base.py
+++ b/python/cuml/test/test_base.py
@@ -20,7 +20,8 @@
 import numpydoc.docscrape
 from cuml.test.utils import (get_classes_from_package,
                              small_classification_dataset)
-from cuml._thirdparty.sklearn.utils.skl_dependencies import sklBaseEstimator
+from cuml._thirdparty.sklearn.utils.skl_dependencies import BaseEstimator \
+                                                            as sklBaseEstimator
 
 all_base_children = get_classes_from_package(cuml, import_sub_packages=True)