1 files changed, 2 insertions, 2 deletions
diff --git a/run.py b/run.py
index cd19277..e1b6a58 100644
--- a/run.py
+++ b/run.py
@@ -176,7 +176,7 @@ elif arguments_mixVideos:
 	print("{} => {}".format(arguments_strFirst, arguments_strSecond))
 	outputPath = './renders/' + arguments_strVideoOut
 	frames = process_two_videos(moduleNetwork, tensorOutput, arguments_strFirst, arguments_strSecond, arguments_aOffset, arguments_bOffset, arguments_steps, arguments_dilate)
-	dilate_frames(moduleNetwork, tensorOutput, frames, arguments_dilate)
+	frames = dilate_frames(moduleNetwork, tensorOutput, frames, arguments_dilate)
 	store_frames(frames, outputPath)
 elif arguments_steps == 0:
 	# Process image
@@ -191,5 +191,5 @@ else:
 	inputSecond = load_image(arguments_strSecond)
 	outputPath = './renders/' + arguments_strVideoOut
 	frames = process_tree(moduleNetwork, inputFirst, inputSecond, tensorOutput, arguments_steps * arguments_dilate, arguments_dilate)
-	dilate_frames(moduleNetwork, tensorOutput, frames, arguments_dilate)
+	frames = dilate_frames(moduleNetwork, tensorOutput, frames, arguments_dilate)
 	store_frames(frames, outputPath, inputFirst, inputSecond)