From 86203c81383358fd2d16a238fe5c79500536362d Mon Sep 17 00:00:00 2001
From: Mike Salvatore <mike.s.salvatore@gmail.com>
Date: Thu, 9 Dec 2021 14:56:54 -0500
Subject: [PATCH] Agent: Add AutomatedMaster to master/__init__.py

---
 monkey/infection_monkey/master/__init__.py                     | 1 +
 .../infection_monkey/master/test_automated_master.py           | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/monkey/infection_monkey/master/__init__.py b/monkey/infection_monkey/master/__init__.py
index e69de29bb..6d3942abd 100644
--- a/monkey/infection_monkey/master/__init__.py
+++ b/monkey/infection_monkey/master/__init__.py
@@ -0,0 +1 @@
+from .automated_master import AutomatedMaster
diff --git a/monkey/tests/unit_tests/infection_monkey/master/test_automated_master.py b/monkey/tests/unit_tests/infection_monkey/master/test_automated_master.py
index 2ae9ae5d4..1610e752b 100644
--- a/monkey/tests/unit_tests/infection_monkey/master/test_automated_master.py
+++ b/monkey/tests/unit_tests/infection_monkey/master/test_automated_master.py
@@ -1,4 +1,5 @@
-from infection_monkey.master.automated_master import AutomatedMaster
+from infection_monkey.master import AutomatedMaster
+
 
 def test_terminate_without_start():
     m = AutomatedMaster(None, None, None)