diff --git a/src/optimise/train.jl b/src/optimise/train.jl
index c7899b08..3bf06d8f 100644
--- a/src/optimise/train.jl
+++ b/src/optimise/train.jl
@@ -39,7 +39,7 @@ function train!(loss, data, opt; cb = () -> ())
       catch e
         if e isa StopException || rethrow(e)
           @info "Stop Condition Met"
-          :stop
+          return :stop
   opt = runall(opt)
   @progress for d in data
     l = loss(d...)