diff --git a/src/kpi_value_writer/Dockerfile b/src/kpi_value_writer/Dockerfile
index c84613cfc33278be1b411899e2f85d02e3fbd97b..70f41128bd8c982f604a3424d2096c918ead080e 100644
--- a/src/kpi_value_writer/Dockerfile
+++ b/src/kpi_value_writer/Dockerfile
@@ -63,6 +63,8 @@ RUN python3 -m pip install -r requirements.txt
 # Add component files into working directory
 WORKDIR /var/teraflow
 COPY src/kpi_value_writer/. kpi_value_writer/
+COPY src/kpi_manager/__init__.py kpi_manager/__init__.py
+COPY src/kpi_manager/client/. kpi_manager/client/
 
 # Start the service
 ENTRYPOINT ["python", "-m", "kpi_value_writer.service"]