summaryrefslogtreecommitdiffstats
path: root/cli
diff options
context:
space:
mode:
Diffstat (limited to 'cli')
-rw-r--r--cli/cli.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/cli/cli.py b/cli/cli.py
index 61db374..21d69ed 100644
--- a/cli/cli.py
+++ b/cli/cli.py
@@ -74,12 +74,12 @@ class OgCLI():
OgDisk.list_disks(self.rest, args[1:])
def set(self, args):
- choices = ['modes']
+ choices = ['modes', 'mode']
parser = argparse.ArgumentParser(prog='ogcli set')
parser.add_argument('item', choices=choices)
parsed_args = parser.parse_args([args[0]])
- if parsed_args.item == 'modes':
+ if parsed_args.item in choices:
OgModes.set_modes(self.rest, args[1:])
def send(self, args):