diff --git a/api/tests/integration_tests/workflow/nodes/code_executor/test_code_jinja2.py b/api/tests/integration_tests/workflow/nodes/code_executor/test_code_jinja2.py
index b65ca88632..c8eb9ec3e4 100644
--- a/api/tests/integration_tests/workflow/nodes/code_executor/test_code_jinja2.py
+++ b/api/tests/integration_tests/workflow/nodes/code_executor/test_code_jinja2.py
@@ -54,14 +54,9 @@ def test_jinja2_template_with_special_characters():
     <pre>'''code block'''</pre>
 </body>
 </html>"""
-    inputs = {
-        "task": {"Task ID": "TASK-123", "Issues": "Line 1\nLine 2\nLine 3"},
-        "status": "completed"
-    }
+    inputs = {"task": {"Task ID": "TASK-123", "Issues": "Line 1\nLine 2\nLine 3"}, "status": "completed"}
 
-    result = CodeExecutor.execute_workflow_code_template(
-        language=CODE_LANGUAGE, code=template, inputs=inputs
-    )
+    result = CodeExecutor.execute_workflow_code_template(language=CODE_LANGUAGE, code=template, inputs=inputs)
 
     # Verify the template rendered correctly with all special characters
     output = result["result"]
@@ -80,9 +75,7 @@ def test_jinja2_template_with_html_textarea_prefill():
     notes_content = "This is a multi-line note.\nWith special chars: 'single' and \"double\" quotes."
     inputs = {"notes": notes_content}
 
-    result = CodeExecutor.execute_workflow_code_template(
-        language=CODE_LANGUAGE, code=template, inputs=inputs
-    )
+    result = CodeExecutor.execute_workflow_code_template(language=CODE_LANGUAGE, code=template, inputs=inputs)
 
     expected_output = f"<textarea name='notes'>{notes_content}</textarea>"
     assert result["result"] == expected_output
diff --git a/api/tests/test_containers_integration_tests/workflow/nodes/code_executor/test_code_jinja2.py b/api/tests/test_containers_integration_tests/workflow/nodes/code_executor/test_code_jinja2.py
index f923379742..ddb079f00c 100644
--- a/api/tests/test_containers_integration_tests/workflow/nodes/code_executor/test_code_jinja2.py
+++ b/api/tests/test_containers_integration_tests/workflow/nodes/code_executor/test_code_jinja2.py
@@ -60,14 +60,9 @@ class TestJinja2CodeExecutor(CodeExecutorTestMixin):
     <pre>'''code block'''</pre>
 </body>
 </html>"""
-        inputs = {
-            "task": {"Task ID": "TASK-123", "Issues": "Line 1\nLine 2\nLine 3"},
-            "status": "completed"
-        }
+        inputs = {"task": {"Task ID": "TASK-123", "Issues": "Line 1\nLine 2\nLine 3"}, "status": "completed"}
 
-        result = CodeExecutor.execute_workflow_code_template(
-            language=CodeLanguage.JINJA2, code=template, inputs=inputs
-        )
+        result = CodeExecutor.execute_workflow_code_template(language=CodeLanguage.JINJA2, code=template, inputs=inputs)
 
         # Verify the template rendered correctly with all special characters
         output = result["result"]