diff --git a/IGEV-Stereo/demo_imgs.py b/IGEV-Stereo/demo_imgs.py
index bddc0e2..ce4067d 100644
--- a/IGEV-Stereo/demo_imgs.py
+++ b/IGEV-Stereo/demo_imgs.py
@@ -59,7 +59,7 @@ def demo(args):
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
     parser.add_argument('--restore_ckpt', help="restore checkpoint", default='./pretrained_models/sceneflow/sceneflow.pth')
-    parser.add_argument('--save_numpy', default=False, help='save output as numpy arrays')
+    parser.add_argument('--save_numpy', action="store_true", help='save output as numpy arrays')
 
     parser.add_argument('-l', '--left_imgs', help="path to all first (left) frames", default="./demo-imgs/*/im0.png")
     parser.add_argument('-r', '--right_imgs', help="path to all second (right) frames", default="./demo-imgs/*/im1.png")