diff --git a/validations_libs/tests/test_ansible.py b/validations_libs/tests/test_ansible.py
index 89225f4..5925892 100644
--- a/validations_libs/tests/test_ansible.py
+++ b/validations_libs/tests/test_ansible.py
@@ -13,7 +13,10 @@
 #   under the License.
 #
 
-from unittest import mock
+try:
+    from unittest import mock
+except ImportError:
+    import mock
 from unittest import TestCase
 
 from ansible_runner import Runner
diff --git a/validations_libs/tests/test_utils.py b/validations_libs/tests/test_utils.py
index fb74853..b1a03be 100644
--- a/validations_libs/tests/test_utils.py
+++ b/validations_libs/tests/test_utils.py
@@ -13,7 +13,10 @@
 #   under the License.
 #
 
-from unittest import mock
+try:
+    from unittest import mock
+except ImportError:
+    import mock
 from unittest import TestCase
 
 from validations_libs import utils
diff --git a/validations_libs/tests/test_validations_list.py b/validations_libs/tests/test_validations_list.py
index 3e30cf0..a5cf5a5 100644
--- a/validations_libs/tests/test_validations_list.py
+++ b/validations_libs/tests/test_validations_list.py
@@ -13,7 +13,10 @@
 #   under the License.
 #
 
-from unittest import mock
+try:
+    from unittest import mock
+except ImportError:
+    import mock
 from unittest import TestCase
 
 from validations_libs.tests import fakes
diff --git a/validations_libs/tests/test_validations_run.py b/validations_libs/tests/test_validations_run.py
index 2749c0d..416a141 100644
--- a/validations_libs/tests/test_validations_run.py
+++ b/validations_libs/tests/test_validations_run.py
@@ -13,7 +13,10 @@
 #   under the License.
 #
 
-from unittest import mock
+try:
+    from unittest import mock
+except ImportError:
+    import mock
 from unittest import TestCase
 
 from validations_libs.tests import fakes
diff --git a/validations_libs/tests/test_validations_show.py b/validations_libs/tests/test_validations_show.py
index 516a20b..9f4a44d 100644
--- a/validations_libs/tests/test_validations_show.py
+++ b/validations_libs/tests/test_validations_show.py
@@ -13,7 +13,10 @@
 #   under the License.
 #
 
-from unittest import mock
+try:
+    from unittest import mock
+except ImportError:
+    import mock
 from unittest import TestCase
 
 from validations_libs.tests import fakes