diff --git a/__init__.py b/__init__.py
index 5c7b2e6..32c0705 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1 +1 @@
-from .spindecoupler import AgentSide,BaselinesSide
+from .spindecoupler import AgentSide,RLSide
diff --git a/spindecoupler.py b/spindecoupler.py
index 96c9ddd..133febb 100644
--- a/spindecoupler.py
+++ b/spindecoupler.py
@@ -8,7 +8,7 @@
 """
 
 from enum import Enum
-from socketcomms.comms import ClientCommPoint,ServerCommPoint
+from .socketcomms.comms import ClientCommPoint,ServerCommPoint
 
 
 #-------------------------------------------------------------------------------