diff --git a/src/optimise/Optimise.jl b/src/optimise/Optimise.jl
index 7da6df70..fd987861 100644
--- a/src/optimise/Optimise.jl
+++ b/src/optimise/Optimise.jl
@@ -1,6 +1,6 @@
 module Optimise
 
-using ..Tracker: TrackedArray, grad, back!
+using ..Tracker: TrackedArray, data, grad, back!
 
 export sgd, update!, params, train!