diff --git a/api/tests/unit_tests/core/workflow/graph_engine/test_graph.py b/api/tests/unit_tests/core/workflow/graph_engine/test_graph.py
index e8e4fc7046..7660cd6ea0 100644
--- a/api/tests/unit_tests/core/workflow/graph_engine/test_graph.py
+++ b/api/tests/unit_tests/core/workflow/graph_engine/test_graph.py
@@ -1,6 +1,4 @@
 from core.workflow.graph_engine.entities.graph import Graph
-from core.workflow.graph_engine.entities.run_condition import RunCondition
-from core.workflow.utils.condition.entities import Condition
 
 
 def test_init():