diff --git a/python/test.py b/python/test.py
index 0ff701f..e51a2ae 100644
--- a/python/test.py
+++ b/python/test.py
@@ -49,6 +49,12 @@ def direction(controller: Controller):
     controller.set_direction(OutAngle.EasyLeft, OutType.LongerLane)
     controller.set_direction(OutAngle.Left, OutType.LongerLane)
     controller.set_direction(OutAngle.SharpLeft, OutType.LongerLane)
+    roundabout(controller)
+    controller.set_direction(OutAngle.Left, OutType.Flag)
+    controller.set_direction(OutAngle.Right, OutType.Flag)
+
+
+def roundabout(controller: Controller):
     controller.set_direction(OutAngle.SharpRight, OutType.RightRoundabout)
     controller.set_direction(OutAngle.Right, OutType.RightRoundabout)
     controller.set_direction(OutAngle.EasyRight, OutType.RightRoundabout)
@@ -56,8 +62,7 @@ def direction(controller: Controller):
     controller.set_direction(OutAngle.EasyLeft, OutType.RightRoundabout)
     controller.set_direction(OutAngle.Left, OutType.RightRoundabout)
     controller.set_direction(OutAngle.SharpLeft, OutType.RightRoundabout)
-    controller.set_direction(OutAngle.Left, OutType.Flag)
-    controller.set_direction(OutAngle.Right, OutType.Flag)
+    controller.set_direction(OutAngle.Down, OutType.RightRoundabout)
 
 
 def distance(controller: Controller):