diff --git a/monkey/common/aws/__init__.py b/monkey/common/aws/__init__.py
index e69de29bb..1bf4dafaa 100644
--- a/monkey/common/aws/__init__.py
+++ b/monkey/common/aws/__init__.py
@@ -0,0 +1 @@
+from .aws_instance import AWSInstance
diff --git a/monkey/infection_monkey/utils/aws_environment_check.py b/monkey/infection_monkey/utils/aws_environment_check.py
index 5e69703e2..dc074cd6a 100644
--- a/monkey/infection_monkey/utils/aws_environment_check.py
+++ b/monkey/infection_monkey/utils/aws_environment_check.py
@@ -1,6 +1,6 @@
 import logging
 
-from common.aws.aws_instance import AWSInstance
+from common.aws import AWSInstance
 from infection_monkey.telemetry.aws_instance_telem import AWSInstanceTelemetry
 from infection_monkey.telemetry.messengers.legacy_telemetry_messenger_adapter import (
     LegacyTelemetryMessengerAdapter,