diff --git a/Project.toml b/Project.toml
index 2351eb66..0372ea4d 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "DynamicalSystems"
 uuid = "61744808-ddfa-5f27-97ff-6e42cc95d634"
 repo = "https://github.com/JuliaDynamics/DynamicalSystems.jl.git"
-version = "3.3.1"
+version = "3.3.2"
 
 [deps]
 Attractors = "f3fd9213-ca85-4dba-9dfd-7fc91308fec7"
diff --git a/ext/src/interactive_trajectory.jl b/ext/src/interactive_trajectory.jl
index 99e06ec8..fd745d3e 100644
--- a/ext/src/interactive_trajectory.jl
+++ b/ext/src/interactive_trajectory.jl
@@ -1,7 +1,5 @@
 using DynamicalSystems.DataStructures
 
-# TODO: make the timeseries interactive the main function.
-
 function DynamicalSystems.interactive_trajectory(
         ds::DynamicalSystems.DynamicalSystem, u0s = [DynamicalSystems.current_state(ds)];
         # Selection of what to plot
@@ -11,7 +9,7 @@ function DynamicalSystems.interactive_trajectory(
         Δt = DynamicalSystems.isdiscretetime(ds) ? 1 : 0.01,
         pause = nothing,
         # Visualization
-        colors = collect(cgrad(COLORSCHEME, N; categorical = true)),
+        colors = collect(cgrad(COLORSCHEME, length(u0s); categorical = true)),
         plotkwargs = NamedTuple(), markersize = 15,
         fade = true,
         # parameters
@@ -254,7 +252,7 @@ function DynamicalSystems.interactive_trajectory_timeseries(
     ds::DynamicalSystem, fs::Vector, u0s = [current_state(ds)];
     linekwargs = isdiscretetime(ds)  ? (linewidth = 1,) : (linewidth = 3,),
     timeseries_names = [timeseries_name(f) for f in fs],
-    colors = collect(cgrad(COLORSCHEME, N; categorical = true)),
+    colors = collect(cgrad(COLORSCHEME, length(u0s); categorical = true)),
     timeseries_ylims = nothing,
     timelabel = "time", timeunit = 1,
     Δt = DynamicalSystems.isdiscretetime(ds) ? 1 : 0.01,