diff --git a/src/batches/Batches.jl b/src/batches/Batches.jl
index 5fc3f862..0146fb00 100644
--- a/src/batches/Batches.jl
+++ b/src/batches/Batches.jl
@@ -2,6 +2,8 @@ module Batches
 
 import ..Flux
 
+export Batch, Tree
+
 include("batch.jl")
 include("tree.jl")