diff --git a/api/tests/unit_tests/core/workflow/nodes/answer/test_answer.py b/api/tests/unit_tests/core/workflow/nodes/answer/test_answer.py
index 2a43b93e2c..0e3faf944c 100644
--- a/api/tests/unit_tests/core/workflow/nodes/answer/test_answer.py
+++ b/api/tests/unit_tests/core/workflow/nodes/answer/test_answer.py
@@ -87,7 +87,7 @@ def test_execute_answer():
 
 
 def test_execute_answer_with_outputs():
-    from core.workflow.enums import SystemVariableKey  
+    from core.workflow.enums import SystemVariableKey
 
     """Test Answer node with custom output variables"""
     graph_config = {
@@ -176,7 +176,7 @@ def test_execute_answer_with_outputs():
 
 
 def test_execute_answer_with_empty_outputs():
-    from core.workflow.enums import SystemVariableKey  
+    from core.workflow.enums import SystemVariableKey
 
     """Test Answer node with empty outputs configuration"""
     graph_config = {