diff --git a/inference_mp4_2x.py b/inference_mp4_2x.py
index 9df0dbd..43c290e 100644
--- a/inference_mp4_2x.py
+++ b/inference_mp4_2x.py
@@ -53,7 +53,7 @@ while success:
             output.write(lastframe)
             output.write(mid[:h, :w])
     if args.montage:
-        output.write(np.concatenate((frame, frame), 1))
+        output.write(np.concatenate((lastframe, lastframe), 1))
     else:
-        output.write(frame)
+        output.write(lastframe)
 output.release()