diff --git a/src/optimise/train.jl b/src/optimise/train.jl
index 4b1e205b..0a06492c 100644
--- a/src/optimise/train.jl
+++ b/src/optimise/train.jl
@@ -1,6 +1,6 @@
 using Juno
 using Flux.Tracker: back!
-import Flux: stop, StopException
+using Flux: stop, StopException
 
 runall(f) = f
 runall(fs::AbstractVector) = () -> foreach(call, fs)