diff --git a/src/Flux.jl b/src/Flux.jl
index 47dd5be3..6288dba7 100644
--- a/src/Flux.jl
+++ b/src/Flux.jl
@@ -22,6 +22,7 @@ import .Tracker: data
 
 include("optimise/Optimise.jl")
 using .Optimise
+using .Optimise: @epochs
 
 include("utils.jl")
 include("onehot.jl")