Minor fixes

SciML · Oct 15, 2024 · 356354f · 356354f
1 parent f831aee
commit 356354f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/preset_time.jl b/src/preset_time.jl
@@ -28,7 +28,7 @@ function PresetTimeCallback(tstops, user_affect!;
         throw(ArgumentError("tstops must either be a number or a Vector. Was $tstops"))
     end
 
-    tstops = sort_inplace ? sort!(tstops) : sort(tstops)
+    tstops = tstops isa Number ? [tstops] : (sort_inplace ? sort!(tstops) : sort(tstops))
 
     condition = let
         function (u, t, integrator)
@@ -57,7 +57,7 @@ function PresetTimeCallback(tstops, user_affect!;
         else
             _tstops = tstops
         end
-        for tstop in tstops
+        for tstop in _tstops
             add_tstop!(integrator, tstop)
         end
         if t ∈ tstops