diff --git a/Trajectories/draw_segments.py b/Trajectories/draw_segments.py
index 6971b3a48b7113bef4646128d4c0446bbcec2811..fa4758c8e0bb87b402513f66e237b5ed21416135 100755
--- a/Trajectories/draw_segments.py
+++ b/Trajectories/draw_segments.py
@@ -49,9 +49,13 @@ if args.t:
         color = next(colors)
 
         for segment in segment_list:
-            n = g_out.get_node(segment)
-            n.attr["label"] = f"{segment} in {i}"
-            n.attr["color"] = color
+            try:
+                n = g_out.get_node(segment)
+            except KeyError:
+                break
+            else:
+                n.attr["label"] = f"{segment} in {i}"
+                n.attr["color"] = color
 
 g_out.write(args.output_file)