diff --git a/.github/workflows/tests_with_comfy.yml b/.github/workflows/tests_with_comfy.yml
index b660e63..0d89b28 100644
--- a/.github/workflows/tests_with_comfy.yml
+++ b/.github/workflows/tests_with_comfy.yml
@@ -23,5 +23,5 @@ jobs:
       - uses: actions/setup-python@v5
         with:
           python-version: '3.11'
-      - run: pip install -r requirements.txt
+      - run: pip install -r requirements.txt -r ComfyUI/requirements.txt
       - run: PYTHONPATH=ComfyUI python -m prompt_control.test_graph
diff --git a/prompt_control/test_graph.py b/prompt_control/test_graph.py
index 01fc3de..8f7c45d 100644
--- a/prompt_control/test_graph.py
+++ b/prompt_control/test_graph.py
@@ -17,7 +17,6 @@ def apply(cls, text, **kwargs):
 
 
 @mock.patch("prompt_control.utils.lora_name_to_file", find_file)
-@mock.patch.dict("sys.modules", nodes=mock.MagicMock())
 class GraphTests(unittest.TestCase):
     maxDiff = 4096