diff --git a/tests/structs/test_flow.py b/tests/structs/test_flow.py
index 84034a08..494833fe 100644
--- a/tests/structs/test_flow.py
+++ b/tests/structs/test_flow.py
@@ -326,7 +326,6 @@ def test_flow_response_filtering(flow_instance):
 def test_flow_undo_last(flow_instance):
     # Test the undo functionality
     response1 = flow_instance.run("Task 1")
-    response2 = flow_instance.run("Task 2")
     previous_state, message = flow_instance.undo_last()
     assert response1 == previous_state
     assert "Restored to" in message
@@ -552,7 +551,6 @@ def test_flow_rollback(flow_instance):
     # Test rolling back to a previous state
     state1 = flow_instance.get_state()
     flow_instance.change_prompt("New prompt")
-    state2 = flow_instance.get_state()
     flow_instance.rollback_to_state(state1)
     assert flow_instance.get_current_prompt() == state1["current_prompt"]
     assert flow_instance.get_instructions() == state1["instructions"]
diff --git a/tests/swarms/test_multi_agent_collab.py b/tests/swarms/test_multi_agent_collab.py
index d5b79fdf..8f140237 100644
--- a/tests/swarms/test_multi_agent_collab.py
+++ b/tests/swarms/test_multi_agent_collab.py
@@ -100,6 +100,7 @@ def test_set_interaction_rules(collaboration):
     assert hasattr(collaboration, "interaction_rules")
     assert collaboration.interaction_rules == rules
 
+
 def test_repr(collaboration):
     repr_str = repr(collaboration)
     assert isinstance(repr_str, str)