diff --git a/demo.py b/demo.py
index 70fb6a3..2f908df 100644
--- a/demo.py
+++ b/demo.py
@@ -502,7 +502,7 @@ def generate_visualization_video(result_path, base_output_dir="results/visualiza
         recon_fun = functools.partial(get_reconstructed_scene, args, tmpdirname, model, args.device, args.silent, args.image_size)
         
         # Call the function with default parameters
-        scene, outfile, imgs = recon_fun(
+        scene, outfile, temp_video_path, temp_attn_video_path, temp_cluster_video_path = recon_fun(
             filelist=input_files,
             schedule='linear',
             niter=300,