diff --git a/model_data/simulation_system/Dockerfiles/Dockerfile.prediction.lambda b/model_data/simulation_system/Dockerfiles/Dockerfile.prediction.lambda
index 0c3db572..bd7479e6 100644
--- a/model_data/simulation_system/Dockerfiles/Dockerfile.prediction.lambda
+++ b/model_data/simulation_system/Dockerfiles/Dockerfile.prediction.lambda
@@ -1,17 +1,20 @@
 FROM public.ecr.aws/lambda/python:3.10
 
 # Set the working directory
-WORKDIR ${LAMBDA_TASK_ROOT}/simulation_system
+WORKDIR ${LAMBDA_TASK_ROOT}
+
+# Install necessary build tools - required to test locally
+# RUN yum install -y gcc python3-devel
 
 # 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" ]
diff --git a/model_data/simulation_system/handlers/predictions_app.py b/model_data/simulation_system/handlers/predictions_app.py
index 1f3a5c8b..02b1673f 100644
--- a/model_data/simulation_system/handlers/predictions_app.py
+++ b/model_data/simulation_system/handlers/predictions_app.py
@@ -1,6 +1,6 @@
 import os
 import urllib.parse
-from predictions import prediction
+from ..predictions import prediction
 
 RUNTIME_ENVIRONMENT = os.environ.get("RUNTIME_ENVIRONMENT", "dev")
 
@@ -30,7 +30,8 @@ def handler(event, context):
     # model_path = os.environ.get("MODEL_PATH", "http://minio:9000/data/model_directory/")
     model_path = os.environ.get(
         "MODEL_PATH",
-        "s3://retrofit-model-directory-{RUNTIME_ENVIRONMENT}/RDSAP_CHANGE/autogluon/rdsap_change-medium_quality-30-2023-08-30_11-43-41/deployment/",
+        "s3://retrofit-model-directory-{RUNTIME_ENVIRONMENT}/RDSAP_CHANGE/autogluon/rdsap_change-medium_quality-30"
+        "-2023-08-30_11-43-41/deployment/",
     )
 
     try:
diff --git a/sapmodel.serverless.yml b/sapmodel.serverless.yml
index 50567b36..a37e2df5 100644
--- a/sapmodel.serverless.yml
+++ b/sapmodel.serverless.yml
@@ -19,7 +19,7 @@ plugins:
 custom:
   customDomain:
     domainName: api.${self:provider.environment.DOMAIN_NAME}
-    basePath: 'v1/sapmodel'
+    basePath: 'sapmodel'
     createRoute53Record: true
     certificateArn: ${ssm:/ssl_certificate_arn}