diff --git a/monkey/infection_monkey/main.py b/monkey/infection_monkey/main.py index fac31bf78..d6edfaec2 100644 --- a/monkey/infection_monkey/main.py +++ b/monkey/infection_monkey/main.py @@ -120,14 +120,12 @@ def main(): monkey = monkey_cls(monkey_args) try: - monkey.legacy_start() - # monkey.start() + monkey.start() return True except Exception as e: logger.exception("Exception thrown from monkey's start function. More info: {}".format(e)) finally: - monkey.legacy_cleanup() - # monkey.cleanup() + monkey.cleanup() if "__main__" == __name__: