diff --git a/model_data/simulation_system/Dockerfiles/Dockerfile.prediction.lambda b/model_data/simulation_system/Dockerfiles/Dockerfile.prediction.lambda
index 0c3db572..512c8150 100644
--- a/model_data/simulation_system/Dockerfiles/Dockerfile.prediction.lambda
+++ b/model_data/simulation_system/Dockerfiles/Dockerfile.prediction.lambda
@@ -1,17 +1,15 @@
-FROM public.ecr.aws/lambda/python:3.10
-
 # Set the working directory
-WORKDIR ${LAMBDA_TASK_ROOT}/simulation_system
+WORKDIR ${LAMBDA_TASK_ROOT}
 
 # Install python packages
-COPY ../requirements/predictions/predictions.txt ./requirements.txt
-RUN pip install --no-cache-dir -r requirements.txt
+COPY requirements/predictions/predictions.txt ./simulation_system/requirements.txt
+RUN pip install --no-cache-dir -r ./simulation_system/requirements.txt
 
 # Copy the project code to the working directory
-COPY ./core ./core
-COPY ./MLModel ./MLModel
-COPY ./predictions.py ./predictions.py
-COPY ./handlers/predictions_app.py ./predictions_app.py
+COPY ./core ./simulation_system/core
+COPY ./MLModel ./simulation_system/MLModel
+COPY ./predictions.py ./simulation_system/predictions.py
+COPY ./handlers/predictions_app.py ./simulation_system/predictions_app.py
 
 # Run off a lambda trigger
-CMD [ "predictions_app.handler" ]
+CMD [ "simulation_system.predictions_app.handler" ]