diff --git a/src/l3_attackmitigator/Dockerfile b/src/l3_attackmitigator/Dockerfile
index 9ecc9bc1216bb5c589cf782dfdb881f178555f1e..a97846982dbeb419dd27f34640c46c2cc1960116 100644
--- a/src/l3_attackmitigator/Dockerfile
+++ b/src/l3_attackmitigator/Dockerfile
@@ -64,6 +64,8 @@ RUN python3 -m pip install -r requirements.txt
 WORKDIR /var/teraflow
 COPY src/l3_attackmitigator/. l3_attackmitigator
 COPY src/monitoring/. monitoring
+COPY src/context/. context/
+COPY src/service/. service/
 
 # Start the service
 ENTRYPOINT ["python", "-m", "l3_attackmitigator.service"]