diff --git a/test/units/mock/loader.py b/test/units/mock/loader.py
index 4bc7eab5a3f..82e6dbac8e2 100644
--- a/test/units/mock/loader.py
+++ b/test/units/mock/loader.py
@@ -30,7 +30,7 @@ class DictDataLoader(DataLoader):
 
     def __init__(self, file_mapping=None):
         file_mapping = {} if file_mapping is None else file_mapping
-        assert type(file_mapping) == dict
+        assert isinstance(file_mapping, dict)
 
         super(DictDataLoader, self).__init__()
 
diff --git a/test/units/module_utils/basic/test_safe_eval.py b/test/units/module_utils/basic/test_safe_eval.py
index e8538ca911d..fdaab18abfc 100644
--- a/test/units/module_utils/basic/test_safe_eval.py
+++ b/test/units/module_utils/basic/test_safe_eval.py
@@ -67,4 +67,4 @@ def test_invalid_strings_with_exceptions(am, code, expected, exception):
     if exception is None:
         assert res[1] == exception
     else:
-        assert type(res[1]) == exception
+        assert isinstance(res[1], exception)