diff --git a/cli/cli.py b/cli/cli.py index a81ed07..81dd846 100644 --- a/cli/cli.py +++ b/cli/cli.py @@ -149,7 +149,7 @@ class OgCLI(): def restore(self, args): choices = ['image'] parser = argparse.ArgumentParser(prog='ogcli restore') - parser.add_argument('send_obj', choices=choices) + parser.add_argument('restore_obj', choices=choices) if not args: print('Missing restore subcommand', file=sys.stderr) @@ -157,7 +157,7 @@ class OgCLI(): sys.exit(1) parsed_args = parser.parse_args([args[0]]) - if parsed_args.send_obj == 'image': + if parsed_args.restore_obj == 'image': OgImage.restore_image(self.rest, args[1:]) def create(self, args):