diff --git a/model_data/simulation_system/Dockerfiles/Dockerfile.prediction.lambda b/model_data/simulation_system/Dockerfiles/Dockerfile.prediction.lambda
index 512c8150..2e301b24 100644
--- a/model_data/simulation_system/Dockerfiles/Dockerfile.prediction.lambda
+++ b/model_data/simulation_system/Dockerfiles/Dockerfile.prediction.lambda
@@ -1,3 +1,5 @@
+FROM public.ecr.aws/lambda/python:3.10
+
 # Set the working directory
 WORKDIR ${LAMBDA_TASK_ROOT}