diff --git a/doc/whats_new/v0.12.rst b/doc/whats_new/v0.12.rst
index 6585c3b58..bc707ae8c 100644
--- a/doc/whats_new/v0.12.rst
+++ b/doc/whats_new/v0.12.rst
@@ -15,6 +15,12 @@ Bug fixes
   `estimator` could not be a :class:`~sklearn.pipeline.Pipeline` object.
   :pr:`1049` by :user:`Gonenc Mogol <gmogol>`.
 
+Compatibility
+.............
+
+- Do not use `distutils` in tests due to deprecation.
+  :pr:`1065` by :user:`Michael R. Crusoe <mr-c>`
+
 Version 0.12.0
 ==============
 
diff --git a/imblearn/tensorflow/tests/test_generator.py b/imblearn/tensorflow/tests/test_generator.py
index bcc10b8f1..e0c7a9103 100644
--- a/imblearn/tensorflow/tests/test_generator.py
+++ b/imblearn/tensorflow/tests/test_generator.py
@@ -1,9 +1,8 @@
-from distutils.version import LooseVersion
-
 import numpy as np
 import pytest
 from scipy import sparse
 from sklearn.datasets import load_iris
+from sklearn.utils.fixes import parse_version
 
 from imblearn.datasets import make_imbalance
 from imblearn.over_sampling import RandomOverSampler
@@ -147,7 +146,7 @@ def accuracy(y_true, y_pred):
 
 @pytest.mark.parametrize("sampler", [None, NearMiss(), RandomOverSampler()])
 def test_balanced_batch_generator(data, sampler):
-    if LooseVersion(tf.__version__) < "2":
+    if parse_version(tf.__version__) < parse_version("2.0.0"):
         check_balanced_batch_generator_tf_1_X_X(data, sampler)
     else:
         check_balanced_batch_generator_tf_2_X_X_compat_1_X_X(data, sampler)