1 files changed, 1 insertions, 0 deletions

diff --git a/run.py b/run.py
index 15ab098..4c4b0ce 100644
--- a/run.py
+++ b/run.py
@@ -142,6 +142,7 @@ def smooth_frames(moduleNetwork, tensorOutput, frames, smooth):
 		new_frames += [firstFrame]
 		new_frames += [middle_np]
 	new_frames += [nextFrame]
+	new_frames = frames[len(frames)-1]
 	return new_frames
 
 def dilate_frames(moduleNetwork, tensorOutput, frames, dilate):