diff --git a/api/tests/unit/plugins/action/test_determine_updates_required.py b/api/tests/unit/plugins/action/test_determine_updates_required.py
index 3fc6cd8..18b10d7 100644
--- a/api/tests/unit/plugins/action/test_determine_updates_required.py
+++ b/api/tests/unit/plugins/action/test_determine_updates_required.py
@@ -1,13 +1,19 @@
 import unittest
 import sys
+from unittest.mock import MagicMock
 
 sys.modules['ansible.utils.display'] = unittest.mock.Mock()
 sys.modules['typing_extensions'] = unittest.mock.Mock()
 
-from .....plugins.action.deploy_target_config import determine_required_updates
+from .....plugins.action.deploy_target_config import ActionModule
 
 class DetermineUpdatesTester(unittest.TestCase):
 
+    def setUp(self):
+        # Setup a mock for the display utility used within ActionModule
+        self.module = ActionModule()
+        self.module._display = MagicMock()
+
     def test_update_required(self):
         existing_configs = []
         desired_configs = [
@@ -18,7 +24,7 @@ def test_update_required(self):
             }
         ]
         
-        addition_required, deletion_required = determine_required_updates(self, existing_configs, desired_configs)
+        addition_required, deletion_required = self.module.determine_required_updates(existing_configs, desired_configs)
         
         assert len(addition_required) == 1, "Expected one addition required"
         assert addition_required[0]['branches'] == '^(main)$', "Expected branches to match ^(main)$"
@@ -44,7 +50,7 @@ def test_update_not_required(self):
             }
         ]
 
-        addition_required, deletion_required = determine_required_updates(self, existing_configs, desired_configs)
+        addition_required, deletion_required = self.module.determine_required_updates(existing_configs, desired_configs)
 
         assert len(addition_required) == 0, "Expected no additions required"
         assert len(deletion_required) == 0, "Expected no deletions required"
@@ -68,7 +74,7 @@ def test_update_required_weight(self):
             }
         ]
 
-        addition_required, deletion_required = determine_required_updates(self, existing_configs, desired_configs)
+        addition_required, deletion_required = self.module.determine_required_updates(existing_configs, desired_configs)
 
         assert len(addition_required) == 1, "Expected one addition required due to weight change"
         assert addition_required[0]['weight'] == 2, "Expected weight to be updated to 2"
@@ -93,7 +99,7 @@ def test_update_required_cluster(self):
             }
         ]
 
-        addition_required, deletion_required = determine_required_updates(self, existing_configs, desired_configs)
+        addition_required, deletion_required = self.module.determine_required_updates(existing_configs, desired_configs)
 
         assert len(addition_required) == 1, "Expected one addition required due to deployTarget change"
         assert addition_required[0]['deployTarget'] == 2, "Expected deployTarget to be updated to 2"
@@ -131,7 +137,7 @@ def test_orphan_existing(self):
             }
         ] 
 
-        addition_required, deletion_required = determine_required_updates(self, existing_configs, desired_configs)
+        addition_required, deletion_required = self.module.determine_required_updates(existing_configs, desired_configs)
 
         assert len(addition_required) == 2, "Expected two additions required due to changes"
         assert addition_required[0]['deployTarget'] == 1, "Expected the first addition to have deployTarget 1"