diff --git a/src/optimise/train.jl b/src/optimise/train.jl
index 1928a80d..92208bc1 100644
--- a/src/optimise/train.jl
+++ b/src/optimise/train.jl
@@ -49,7 +49,7 @@ function train!(loss, data, opt; cb = () -> ())
       else
         rethrow(ex)
       end
-
+    end
   end
 end