diff --git a/examples/UNet/massachusetts_road_segm.py b/examples/UNet/massachusetts_road_segm.py
index f26a2b4..73fd281 100644
--- a/examples/UNet/massachusetts_road_segm.py
+++ b/examples/UNet/massachusetts_road_segm.py
@@ -124,7 +124,8 @@ def main():
     updates = lasagne.updates.adam(loss, params, learning_rate=learning_rate)
 
     # create a convenience function to get the segmentation
-    seg_output = lasagne.layers.get_output(net["output_segmentation"], x_sym, deterministic=True)
+
+    seg_output = lasagne.layers.get_output(net["output_segmentation"], x_sym, deterministic=True, batch_norm_update_averages=False, batch_norm_use_averages=False)
     seg_output = seg_output.argmax(1)
 
     train_fn = theano.function([x_sym, seg_sym, w_sym], [loss, acc_train], updates=updates)