diff --git a/src/tests/ofc22/Dockerfile b/src/tests/ofc22/Dockerfile
index ae2e7d8e90dfbb1414d1220b0c6e722ccdcefdb2..0c4bba9005901715a53e91d54f18c74f156e14ef 100644
--- a/src/tests/ofc22/Dockerfile
+++ b/src/tests/ofc22/Dockerfile
@@ -62,12 +62,21 @@ RUN python3 -m pip install -r requirements.txt
 
 # Add component files into working directory
 WORKDIR /var/teraflow
-COPY src/common/. ./common/
-COPY src/tests/ofc22/tests/. ./tests/ofc22/tests/
 COPY src/__init__.py ./__init__.py
-COPY src/tests/__init__.py ./tests/__init__.py
+COPY src/common/*.py ./common/
+COPY src/common/tests/. ./common/tests/
+COPY src/common/tools/. ./common/tools/
+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/monitoring/__init__.py monitoring/__init__.py
+COPY src/monitoring/client/. monitoring/client/
+COPY src/tests/*.py ./tests/
 COPY src/tests/ofc22/__init__.py ./tests/ofc22/__init__.py
 COPY src/tests/ofc22/descriptors_emulated.json ./tests/ofc22/descriptors_emulated.json
+COPY src/tests/ofc22/tests/. ./tests/ofc22/tests/
+COPY src/tests/tools/. ./tests/tools/
 
 RUN tee ./run_tests.sh <<EOF
 #!/bin/bash