From 793bb33c8cd780ed39c38d05ad8d6a68be41d794 Mon Sep 17 00:00:00 2001
From: VakarisZ <vakarisz@yahoo.com>
Date: Wed, 1 Dec 2021 17:04:45 +0200
Subject: [PATCH] Agent: use the refactored startup instead of legacy methods
 (monkey.start() instead of monkey.legacy_start(), etc.)

---
 monkey/infection_monkey/main.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

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__: