diff --git a/plugins/socksproxyserver.py b/plugins/socksproxyserver.py
index 7b27925..b45307e 100644
--- a/plugins/socksproxyserver.py
+++ b/plugins/socksproxyserver.py
@@ -1,7 +1,12 @@
 from __future__ import print_function
+import importlib
 
-from lib.common.plugins import Plugin
-import lib.common.helpers as helpers
+if importlib.util.find_spec("empire") is not None:
+    from empire.server.common.plugins import Plugin
+    import empire.server.common.helpers as helpers    
+else:
+    from lib.common.plugins import Plugin
+    import lib.common.helpers as helpers    
 
 import socket
 import _thread