diff --git a/monkey/infection_monkey/puppet/puppet.py b/monkey/infection_monkey/puppet/puppet.py
index 3e2ad0f5f..41ed99250 100644
--- a/monkey/infection_monkey/puppet/puppet.py
+++ b/monkey/infection_monkey/puppet/puppet.py
@@ -11,9 +11,9 @@ from infection_monkey.i_puppet import (
     PortScanData,
     PostBreachData,
 )
-from infection_monkey.puppet.plugin_registry import PluginRegistry
 
 from .mock_puppet import MockPuppet
+from .plugin_registry import PluginRegistry
 
 logger = logging.getLogger()