diff --git a/b_asic/schedule.py b/b_asic/schedule.py
index 3e8d7201f64bd9d3dc637bc04f0af66247f1618d..ae41ba0b21867ddc9c5bc0f32dccf0787907a5e0 100644
--- a/b_asic/schedule.py
+++ b/b_asic/schedule.py
@@ -420,9 +420,17 @@ class Schedule:
                         [start[0], (start[1] + end[1]) / 2],
                         [start[0] - 0.2, (start[1] + end[1]) / 2],
                         [start[0] - 0.2, end[1]],
-                        end
+                        end,
+                    ],
+                    [
+                        Path.MOVETO,
+                        Path.CURVE4,
+                        Path.CURVE4,
+                        Path.CURVE4,
+                        Path.CURVE4,
+                        Path.CURVE4,
+                        Path.CURVE4,
                     ],
-                    [Path.MOVETO, Path.CURVE4, Path.CURVE4, Path.CURVE4, Path.CURVE4, Path.CURVE4, Path.CURVE4]
                 )
                 pp = PathPatch(p, fc='none')
                 ax.add_patch(pp)
@@ -434,12 +442,11 @@ class Schedule:
                         [(start[0] + end[0]) / 2, end[1]],
                         end,
                     ],
-                    [Path.MOVETO, Path.CURVE4, Path.CURVE4, Path.CURVE4]
-                    )
+                    [Path.MOVETO, Path.CURVE4, Path.CURVE4, Path.CURVE4],
+                )
                 pp = PathPatch(p, fc='none')
                 ax.add_patch(pp)
 
-
         def _draw_offset_arrow(
             start, end, start_offset, end_offset, name="", laps=0
         ):