diff --git a/src/core/tests/Test_CoreMain.py b/src/core/tests/Test_CoreMain.py
index daee38d3..e25f5e99 100644
--- a/src/core/tests/Test_CoreMain.py
+++ b/src/core/tests/Test_CoreMain.py
@@ -1004,8 +1004,8 @@ def test_auto_assessment_resets_maintenance_run_id_and_health_store_id_to_None(s
         CoreMain(argument_composer.get_composed_arguments())
 
         # check values of health_store_id and maintenance_run_id
-        self.assertEquals(runtime.execution_config.health_store_id, None)
-        self.assertEquals(runtime.execution_config.maintenance_run_id, None)
+        self.assertEqual(runtime.execution_config.health_store_id, None)
+        self.assertEqual(runtime.execution_config.maintenance_run_id, None)
 
         # check telemetry events
         self.__check_telemetry_events(runtime)
diff --git a/src/core/tests/Test_UbuntuProClient.py b/src/core/tests/Test_UbuntuProClient.py
index ed169b31..9865df00 100644
--- a/src/core/tests/Test_UbuntuProClient.py
+++ b/src/core/tests/Test_UbuntuProClient.py
@@ -13,10 +13,11 @@
 # limitations under the License.
 #
 # Requires Python 2.7+
-import imp
 import sys
 import types
 import unittest
+if sys.version_info < (3, 12):
+    import imp
 
 from core.src.bootstrap.Constants import Constants
 from core.tests.library.ArgumentComposer import ArgumentComposer
@@ -113,7 +114,7 @@ def __init__(self, package, version, provided_by, origin):
 
 class MockUpdatesResult(MockSystemModules):
 
-    def __init__(self, updates = []):
+    def __init__(self, updates=[]):
         self.updates = updates
 
     def mock_update_list_with_all_update_types(self):
diff --git a/src/extension/tests/Test_ExtEnvHandler.py b/src/extension/tests/Test_ExtEnvHandler.py
index d650240d..bc8c1901 100644
--- a/src/extension/tests/Test_ExtEnvHandler.py
+++ b/src/extension/tests/Test_ExtEnvHandler.py
@@ -233,7 +233,7 @@ def test_get_temp_folder_success(self):
         temp_folder_path = ext_env_handler.get_temp_folder()
 
         # validate path
-        self.assertEquals(temp_folder_path, ext_env_handler.temp_folder)
+        self.assertEqual(temp_folder_path, ext_env_handler.temp_folder)
 
         shutil.rmtree(test_dir)