diff --git a/src/pathcomp/frontend/Dockerfile b/src/pathcomp/frontend/Dockerfile
index 4f694c79a7a2f90fedca21d96f8fe324a0e3b77a..c2b9a7a7c7892a3dbce4ed1295b1530e292751c6 100644
--- a/src/pathcomp/frontend/Dockerfile
+++ b/src/pathcomp/frontend/Dockerfile
@@ -68,10 +68,9 @@ COPY src/context/__init__.py context/__init__.py
 COPY src/context/client/. context/client/
 COPY src/device/__init__.py device/__init__.py
 COPY src/device/client/. device/client/
-COPY src/forecaster/__init__.py forecaster/__init__.py
-COPY src/forecaster/client/. forecaster/client/
-COPY src/forecaster/service/. forecaster/service/
-COPY src/forecaster/tests/. forecaster/tests/
+# Requires entire Forecaster component for the tests
+# TODO: Improve testing framework
+COPY src/forecaster/. forecaster/
 COPY src/monitoring/__init__.py monitoring/__init__.py
 COPY src/monitoring/client/. monitoring/client/
 COPY src/service/__init__.py service/__init__.py