1 files changed, 1 insertions, 1 deletions
diff --git a/run.py b/run.py
index c705552..77be351 100644
--- a/run.py
+++ b/run.py
@@ -54,7 +54,7 @@ def process_tree(moduleNetwork, a, b, tensorOutput, steps):
 	tensorInputSecond = torch.FloatTensor(b)
 	process(moduleNetwork, tensorInputFirst, tensorInputSecond, tensorOutput)
 	middle = tensorOutput.clamp(0.0, 1.0).numpy()
-	np_middle = (numpy.rollaxis(output, 0, 3)[:,:,::-1] * 255.0).astype(numpy.uint8)
+	np_middle = (numpy.rollaxis(middle, 0, 3)[:,:,::-1] * 255.0).astype(numpy.uint8)
 	if steps < 2:
 		return [np_middle]
 	else: