diff --git a/src/training.jl b/src/training.jl
index 3c564ed4..70088444 100644
--- a/src/training.jl
+++ b/src/training.jl
@@ -1,4 +1,5 @@
 using Juno: info
+using .Batches: tobatch
 
 """
   @cb for ... end t expr