diff --git a/terraform_compliance/common/error_handling.py b/terraform_compliance/common/error_handling.py
index 6068cfe2..b72112bb 100644
--- a/terraform_compliance/common/error_handling.py
+++ b/terraform_compliance/common/error_handling.py
@@ -4,7 +4,7 @@
 from terraform_compliance.main import Step
 import colorful
 from ast import literal_eval
-from mock import MagicMock
+from unittest.mock import MagicMock
 import os
 
 
diff --git a/tests/terraform_compliance/steps/test_main_steps.py b/tests/terraform_compliance/steps/test_main_steps.py
index 0d79baf0..cbc95b7b 100644
--- a/tests/terraform_compliance/steps/test_main_steps.py
+++ b/tests/terraform_compliance/steps/test_main_steps.py
@@ -16,7 +16,7 @@
 )
 from terraform_compliance.common.exceptions import TerraformComplianceNotImplemented, Failure
 from tests.mocks import MockedStep, MockedWorld
-from mock import patch
+from unittest.mock import patch
 
 
 class TestStepCases(TestCase):