Fixed example codecomplete_flag_names

1 files changed, 5 insertions, 2 deletions

diff --git a/examples/help_categories.py b/examples/help_categories.py
index c5c9af86..b2a1623f 100755
--- a/examples/help_categories.py
+++ b/examples/help_categories.py
@@ -10,6 +10,7 @@ import functools
 import cmd2
 from cmd2 import (
     COMMAND_NAME,
+    argparse_custom,
 )
 
 
@@ -55,7 +56,9 @@ class HelpCategories(cmd2.Cmd):
         """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 @@ class HelpCategories(cmd2.Cmd):
         """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')