diff --git a/api/tests/test_custom_tools.py b/api/tests/test_custom_tools.py
index 0b1f5a77..186deb44 100644
--- a/api/tests/test_custom_tools.py
+++ b/api/tests/test_custom_tools.py
@@ -1201,6 +1201,9 @@ class TestCustomToolManagerUnit:
             mock_engine
         )
         mock_engine.llm = mock_llm
+        mock_engine._perform_variable_extraction_if_needed = AsyncMock()
+        mock_engine._current_node = None
+        mock_engine._gathered_context = {}
 
         manager = CustomToolManager(mock_engine)