diff --git a/src/common/method_wrappers/tests/deploy_specs.sh b/src/common/method_wrappers/tests/deploy_specs.sh
index ab90ab3d3e7f7b32be59041e5f403fba53f98792..a5af70b04a84ffa83b0e19da005175f0291c4f93 100644
--- a/src/common/method_wrappers/tests/deploy_specs.sh
+++ b/src/common/method_wrappers/tests/deploy_specs.sh
@@ -7,7 +7,7 @@ export TFS_REGISTRY_IMAGE="http://localhost:32000/tfs/"
 #   interdomain slice pathcomp dlt
 #   dbscanserving opticalattackmitigator opticalattackdetector
 #   l3_attackmitigator l3_centralizedattackdetector l3_distributedattackdetector
-export TFS_COMPONENTS="context device pathcomp service slice webui load_generator" # automation monitoring compute
+export TFS_COMPONENTS="context device pathcomp service slice webui load_generator" # automation monitoring compute dlt
 
 # Set the tag you want to use for your images.
 export TFS_IMAGE_TAG="dev"
diff --git a/src/webui/Dockerfile b/src/webui/Dockerfile
index a17d2bd9aea9c6948262dcf17776f75c0be351b8..f0ab356290dd219dd4caa9a0a06251907786323c 100644
--- a/src/webui/Dockerfile
+++ b/src/webui/Dockerfile
@@ -77,6 +77,8 @@ COPY --chown=webui:webui src/context/__init__.py context/__init__.py
 COPY --chown=webui:webui src/context/client/. context/client/
 COPY --chown=webui:webui src/device/__init__.py device/__init__.py
 COPY --chown=webui:webui src/device/client/. device/client/
+COPY --chown=webui:webui src/load_generator/__init__.py load_generator/__init__.py
+COPY --chown=webui:webui src/load_generator/client/. load_generator/client/
 COPY --chown=webui:webui src/service/__init__.py service/__init__.py
 COPY --chown=webui:webui src/service/client/. service/client/
 COPY --chown=webui:webui src/slice/__init__.py slice/__init__.py