diff --git a/main.py b/main.py
index 1e7865d..fd5cf0b 100755
--- a/main.py
+++ b/main.py
@@ -94,8 +94,8 @@ def main():
             turn_degree_v2(Ctrl, msg, degree=-90, absolute=True)
             go_until_yellow_area(Ctrl, msg)
         elif TASK == TaskType.RED_BAR:
-            from task_3.task_3 import go_until_red_area
-            go_until_red_area(Ctrl, msg)
+            from task_4.task_4 import go_straight_until_red_bar
+            go_straight_until_red_bar(Ctrl, msg)
         elif TASK == TaskType.UP_AND_DOWN:
             from task_3.task_3 import go_straight_with_enhanced_calibration
             go_straight_with_enhanced_calibration(Ctrl, msg, distance = 5, speed=0.5, observe=False, mode=11, gait_id=3, step_height=[0.21, 0.21])