diff --git a/Trajectories/Analysis/plot_traj.py b/Trajectories/Analysis/plot_traj.py
index 25693ba972935fb5be6c165e338c1d8d1d40d770..4b46311635edb18f5946a3301d8cf99bb657b83b 100755
--- a/Trajectories/Analysis/plot_traj.py
+++ b/Trajectories/Analysis/plot_traj.py
@@ -214,9 +214,11 @@ if __name__ == "__main__":
                     "cyan",
                 ]
             )
+            arrows = True
         else:
             color = "red"
             color_initial = "green"
+            arrows = False
 
         if args.min_duration == 1:
             for traj in expanded_traj["traj"]:
@@ -234,7 +236,7 @@ if __name__ == "__main__":
                     args.annotate,
                     color,
                     color_initial,
-                    arrows=False,
+                    arrows,
                 )
         else:
             # args.min_duration > 1
@@ -260,7 +262,7 @@ if __name__ == "__main__":
                         args.annotate,
                         color,
                         color_initial,
-                        arrows=False,
+                        arrows,
                     )
 
             print(