From 7701aa38b2e7a79b7c46fccda76dae2a796dee3f Mon Sep 17 00:00:00 2001
From: Stella <30465823+stellaprins@users.noreply.github.com>
Date: Thu, 6 Feb 2025 10:26:40 +0000
Subject: [PATCH] improve code coverage

---
 movement/plot.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/movement/plot.py b/movement/plot.py
index 62aa3583..dd9d213c 100644
--- a/movement/plot.py
+++ b/movement/plot.py
@@ -105,11 +105,10 @@ def trajectory(
     for key, value in DEFAULT_PLOTTING_ARGS.items():
         kwargs.setdefault(key, value)
 
+    colorbar = False
     if "c" not in kwargs:
         kwargs["c"] = plot_point.time
         colorbar = True
-    else:
-        colorbar = False
 
     # Plot the scatter, coloring by time or user-provided color
     sc = ax.scatter(