diff --git a/codes/process_video.py b/codes/process_video.py
index 056f57fa..24a6181e 100644
--- a/codes/process_video.py
+++ b/codes/process_video.py
@@ -161,11 +161,8 @@ if __name__ == "__main__":
 
         model.feed_data(data, need_GT=need_GT)
         model.test()
+        visuals = model.get_current_visuals()['rlt']
 
-        if isinstance(model.fake_H, tuple):
-            visuals = model.fake_H[img_index].detach()
-        else:
-            visuals = model.fake_H.detach()
         if recurrent_mode:
             recurrent_entry = visuals
         visuals = visuals.cpu().float()