diff --git a/api/tests/unit_tests/core/workflow/graph_engine/graph_traversal/__init__.py b/api/tests/unit_tests/core/workflow/graph_engine/graph_traversal/__init__.py
index 76f74616f7a..cf8811dc2b5 100644
--- a/api/tests/unit_tests/core/workflow/graph_engine/graph_traversal/__init__.py
+++ b/api/tests/unit_tests/core/workflow/graph_engine/graph_traversal/__init__.py
@@ -1 +1 @@
-"""Tests for graph traversal components."""
\ No newline at end of file
+"""Tests for graph traversal components."""
diff --git a/api/tests/unit_tests/core/workflow/graph_engine/graph_traversal/test_skip_propagator.py b/api/tests/unit_tests/core/workflow/graph_engine/graph_traversal/test_skip_propagator.py
index fedc514c41f..0019020ede9 100644
--- a/api/tests/unit_tests/core/workflow/graph_engine/graph_traversal/test_skip_propagator.py
+++ b/api/tests/unit_tests/core/workflow/graph_engine/graph_traversal/test_skip_propagator.py
@@ -2,9 +2,6 @@
 
 from unittest.mock import MagicMock, create_autospec
 
-import pytest
-
-from core.workflow.enums import NodeState
 from core.workflow.graph import Edge, Graph
 from core.workflow.graph_engine.graph_state_manager import GraphStateManager
 from core.workflow.graph_engine.graph_traversal.skip_propagator import SkipPropagator
@@ -307,4 +304,4 @@ class TestSkipPropagator:
         propagator.propagate_skip_from_edge("edge_1")
 
         # Assert - should propagate skip
-        mock_state_manager.mark_node_skipped.assert_called_once_with("node_2")
\ No newline at end of file
+        mock_state_manager.mark_node_skipped.assert_called_once_with("node_2")