Fixed example code

python-cmd2 · Sep 7, 2021 · 4c5bfc0 · 4c5bfc0
1 parent 967a8ba
commit 4c5bfc0
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/examples/help_categories.py b/examples/help_categories.py
@@ -10,6 +10,7 @@
 import cmd2
 from cmd2 import (
     COMMAND_NAME,
+    argparse_custom,
 )
 
 
@@ -55,7 +56,9 @@ def do_deploy(self, _):
         """Deploy command"""
         self.poutput('Deploy')
 
-    start_parser = cmd2.DEFAULT_ARGUMENT_PARSER(description='Start', epilog='my_decorator runs even with argparse errors')
+    start_parser = argparse_custom.DEFAULT_ARGUMENT_PARSER(
+        description='Start', epilog='my_decorator runs even with argparse errors'
+    )
     start_parser.add_argument('when', choices=START_TIMES, help='Specify when to start')
 
     @my_decorator
@@ -72,7 +75,7 @@ def do_redeploy(self, _):
         """Redeploy command"""
         self.poutput('Redeploy')
 
-    restart_parser = cmd2.DEFAULT_ARGUMENT_PARSER(
+    restart_parser = argparse_custom.DEFAULT_ARGUMENT_PARSER(
         description='Restart', epilog='my_decorator does not run when argparse errors'
     )
     restart_parser.add_argument('when', choices=START_TIMES, help='Specify when to restart')