diff --git a/accel_sdxl_gen_img.py b/accel_sdxl_gen_img.py
index 79453cfd..bd651c9b 100644
--- a/accel_sdxl_gen_img.py
+++ b/accel_sdxl_gen_img.py
@@ -1846,8 +1846,8 @@ def main(args):
     pipe.set_control_nets(control_nets)
     logger.info(f"pipeline on {device} is ready.")
     distributed_state.wait_for_everyone()
-    pipes = gather_objects([pipe])
-    unets = gather_objects([unet])
+    pipes = gather_object([pipe])
+    unets = gather_object([unet])
 
     if args.diffusers_xformers:
         pipe.enable_xformers_memory_efficient_attention()