diff --git a/docker/Dockerfile.agent b/docker/Dockerfile.agent
index 1fcb8c3b..1e9403e4 100644
--- a/docker/Dockerfile.agent
+++ b/docker/Dockerfile.agent
@@ -43,6 +43,8 @@ RUN if [ "$WITH_EMBEDDINGS" = "true" ]; then \
 COPY backend/src/__init__.py /app/backend/src/__init__.py
 COPY backend/src/core/__init__.py /app/backend/src/core/__init__.py
 COPY backend/src/core/cot_logger.py /app/backend/src/core/cot_logger.py
+COPY backend/src/core/logger.py /app/backend/src/core/logger.py
+COPY backend/src/core/ws_log_handler.py /app/backend/src/core/ws_log_handler.py
 COPY backend/src/agent/ /app/backend/src/agent/
 
 # Gradio server