diff --git a/agents/__init__.py b/agents/__init__.py
index 225b0cc..8d8b1c3 100644
--- a/agents/__init__.py
+++ b/agents/__init__.py
@@ -10,11 +10,9 @@
 from .agent import Agent
 from .chaos import AgentOfChaos
 from .mute import MuteAgent
-from .remote import RemoteAgent
 
 __all__ = [
     "Agent",
     "AgentOfChaos",
     "MuteAgent",
-    "RemoteAgent",
 ]
diff --git a/pyserver/Containerfile b/pyserver/Containerfile
index c9d100d..aa0af6f 100644
--- a/pyserver/Containerfile
+++ b/pyserver/Containerfile
@@ -19,6 +19,7 @@ RUN pip install --no-cache-dir --no-index --find-links=/wheels \
         -r requirements-pyserver.txt && rm -rf /wheels
 
 # Install PyServer code
+COPY agents/ agents/
 COPY pyserver/ pyserver/
 COPY server.py .