diff --git a/tests/unit/test_base/test_flow_parameter_specs.py b/tests/unit/test_base/test_flow_parameter_specs.py
index db811711..0a3c3354 100644
--- a/tests/unit/test_base/test_flow_parameter_specs.py
+++ b/tests/unit/test_base/test_flow_parameter_specs.py
@@ -11,14 +11,6 @@ from trustgraph.base.flow_processor import FlowProcessor
 from trustgraph.base import ParameterSpec, ConsumerSpec, ProducerSpec
 
 
-def mock_async_processor_init(self, **params):
-    """Mock AsyncProcessor.__init__ that properly initializes required attributes"""
-    self.config_handlers = []
-    self.id = params.get('id', 'test-service')
-
-
-# Apply the mock globally for this test module
-patch('trustgraph.base.async_processor.AsyncProcessor.__init__', mock_async_processor_init).start()
 
 
 class TestFlowParameterSpecs(IsolatedAsyncioTestCase):
diff --git a/tests/unit/test_base/test_llm_service_parameters.py b/tests/unit/test_base/test_llm_service_parameters.py
index 895748fb..847e7400 100644
--- a/tests/unit/test_base/test_llm_service_parameters.py
+++ b/tests/unit/test_base/test_llm_service_parameters.py
@@ -12,21 +12,22 @@ from trustgraph.base import ParameterSpec, ConsumerSpec, ProducerSpec
 from trustgraph.schema import TextCompletionRequest, TextCompletionResponse
 
 
-def mock_async_processor_init(self, **params):
-    """Mock AsyncProcessor.__init__ that properly initializes required attributes"""
-    self.config_handlers = []
-    self.id = params.get('id', 'test-service')
-
-
-# Apply the mock globally for this test module
-patch('trustgraph.base.async_processor.AsyncProcessor.__init__', mock_async_processor_init).start()
 
 
 class TestLlmServiceParameters(IsolatedAsyncioTestCase):
     """Test LLM service parameter specification functionality"""
 
-    def test_parameter_specs_registration(self):
+    @patch('trustgraph.base.async_processor.AsyncProcessor.__init__')
+    def test_parameter_specs_registration(self, mock_async_init):
         """Test that LLM service registers model and temperature parameter specs"""
+        # Mock AsyncProcessor.__init__ to set required attributes
+        def mock_init(self, **params):
+            self.config_handlers = []
+            self.id = params.get('id', 'test-service')
+            self.specifications = []
+
+        mock_async_init.side_effect = mock_init
+
         # Arrange
         config = {
             'id': 'test-llm-service',
@@ -44,7 +45,8 @@ class TestLlmServiceParameters(IsolatedAsyncioTestCase):
         assert "temperature" in param_specs
         assert len(param_specs) >= 2  # May have other parameter specs
 
-    def test_model_parameter_spec_properties(self):
+    @patch('trustgraph.base.async_processor.AsyncProcessor.__init__', return_value=None)
+    def test_model_parameter_spec_properties(self, mock_async_init):
         """Test that model parameter spec has correct properties"""
         # Arrange
         config = {
@@ -54,6 +56,9 @@ class TestLlmServiceParameters(IsolatedAsyncioTestCase):
 
         # Act
         service = LlmService(**config)
+        # Manually set required attributes that would be set by AsyncProcessor
+        service.config_handlers = []
+        service.id = config['id']
 
         # Assert
         model_spec = None
@@ -65,7 +70,8 @@ class TestLlmServiceParameters(IsolatedAsyncioTestCase):
         assert model_spec is not None
         assert model_spec.name == "model"
 
-    def test_temperature_parameter_spec_properties(self):
+    @patch('trustgraph.base.async_processor.AsyncProcessor.__init__', return_value=None)
+    def test_temperature_parameter_spec_properties(self, mock_async_init):
         """Test that temperature parameter spec has correct properties"""
         # Arrange
         config = {
@@ -75,6 +81,9 @@ class TestLlmServiceParameters(IsolatedAsyncioTestCase):
 
         # Act
         service = LlmService(**config)
+        # Manually set required attributes that would be set by AsyncProcessor
+        service.config_handlers = []
+        service.id = config['id']
 
         # Assert
         temperature_spec = None