diff --git a/src/backend/tensorflow/model.jl b/src/backend/tensorflow/model.jl
index 63dbc576..ed869504 100644
--- a/src/backend/tensorflow/model.jl
+++ b/src/backend/tensorflow/model.jl
@@ -54,7 +54,7 @@ function Flux.train!(m::Model, train, test=[]; epoch = 1, η = 0.1,
   for e in 1:epoch
     info("Epoch $e\n")
     @progress for (x, y) in train
-      y, cur_loss, _ = run(m.session, vcat(m.output, Loss, minimize_op),
+      y, cur_loss, _ = run(m.session, [m.output, Loss, minimize_op],
                            Dict(m.inputs[1] => batchone(convertel(Float32, x)),
                                 Y => batchone(convertel(Float32, y))))
       if i % 5000 == 0