diff --git a/src/compute/Dockerfile b/src/compute/Dockerfile
index bdc07584c5bd8f08fdef6f997cc18dcfd9eeb3e6..90a69c0f503724fd1098608d85ad5eca874e3f8b 100644
--- a/src/compute/Dockerfile
+++ b/src/compute/Dockerfile
@@ -66,6 +66,8 @@ COPY src/compute/. compute/
 COPY src/context/. context/
 COPY src/service/. service/
 COPY src/slice/. slice/
+RUN mkdir -p /var/teraflow/tests/tools
+COPY src/tests/tools/mock_osm/. tests/tools/mock_osm/
 
 # Start the service
 ENTRYPOINT ["python", "-m", "compute.service"]