diff --git a/src/Batches/Batches.jl b/src/Batches/Batches.jl
index 0cd77b16..7ac656a9 100644
--- a/src/Batches/Batches.jl
+++ b/src/Batches/Batches.jl
@@ -1,6 +1,7 @@
 module Batches
 
 using Juno, Lazy
+using Juno: Tree, Row
 
 export CatMat, rawbatch,
   Batch, Batched, batchone, tobatch, rebatch,
diff --git a/src/Flux.jl b/src/Flux.jl
index abe4c24b..9a508002 100644
--- a/src/Flux.jl
+++ b/src/Flux.jl
@@ -8,7 +8,6 @@ using DataFlow: graphm, syntax, prewalk!, postwalk!, prewalk, postwalk,
   detuple, value, inputs, thread!, value, inputs, Split, splitnode, inputnode,
   spliceinputs, bumpinputs, Line, Frame, applylines, graphinputs
 using DataFlow.Interpreter
-using Juno: Tree, Row
 
 export @net, unroll, unroll1, @shapes,
   @Chain, Chain, Input, Affine, Conv2D, Recurrent, GatedRecurrent, LSTM,