diff --git a/monkey/infection_monkey/dropper.py b/monkey/infection_monkey/dropper.py index 48e5bb86d..3ba310492 100644 --- a/monkey/infection_monkey/dropper.py +++ b/monkey/infection_monkey/dropper.py @@ -46,7 +46,7 @@ class MonkeyDrops(object): arg_parser.add_argument("-p", "--parent") arg_parser.add_argument("-t", "--tunnel") arg_parser.add_argument("-s", "--server") - arg_parser.add_argument("-d", "--depth", type=positive_int) + arg_parser.add_argument("-d", "--depth", type=positive_int, default=0) arg_parser.add_argument("-l", "--location") arg_parser.add_argument("-vp", "--vulnerable-port") self.opts = arg_parser.parse_args(args) diff --git a/monkey/infection_monkey/monkey.py b/monkey/infection_monkey/monkey.py index e02e4e2eb..d91f28018 100644 --- a/monkey/infection_monkey/monkey.py +++ b/monkey/infection_monkey/monkey.py @@ -106,7 +106,7 @@ class InfectionMonkey: arg_parser.add_argument("-p", "--parent") arg_parser.add_argument("-t", "--tunnel") arg_parser.add_argument("-s", "--server") - arg_parser.add_argument("-d", "--depth", type=positive_int) + arg_parser.add_argument("-d", "--depth", type=positive_int, default=0) opts = arg_parser.parse_args(args) InfectionMonkey._log_arguments(opts)