diff --git a/Matrix-Game-3/pipeline/inference_interactive_pipeline.py b/Matrix-Game-3/pipeline/inference_interactive_pipeline.py
index afc2da9..abfa045 100644
--- a/Matrix-Game-3/pipeline/inference_interactive_pipeline.py
+++ b/Matrix-Game-3/pipeline/inference_interactive_pipeline.py
@@ -803,7 +803,4 @@ def to_numpy(x):
                 else:
                     video = None
 
-            if dist.is_initialized():
-                dist.barrier()
-                dist.destroy_process_group()
-            exit()
\ No newline at end of file
+            return video
diff --git a/Matrix-Game-3/pipeline/inference_pipeline.py b/Matrix-Game-3/pipeline/inference_pipeline.py
index 0a0a9a5..abb1693 100644
--- a/Matrix-Game-3/pipeline/inference_pipeline.py
+++ b/Matrix-Game-3/pipeline/inference_pipeline.py
@@ -706,7 +706,4 @@ def to_numpy(x):
             else:
                 video = None
 
-            if dist.is_initialized():
-                dist.barrier()
-                dist.destroy_process_group()
-            exit()
\ No newline at end of file
+            return video