diff --git a/backend/docker/lambda.Dockerfile b/backend/docker/lambda.Dockerfile
index db9bad40..168f94f6 100644
--- a/backend/docker/lambda.Dockerfile
+++ b/backend/docker/lambda.Dockerfile
@@ -1,30 +1,30 @@
-# Some additional work is required to get mip to run on arm64 architecture.
-# This is because the mip library relies on the CBC solver, which is not available for arm64.
-# https://python-mip.readthedocs.io/en/latest/install.html
-
 # Pull base image
-FROM arm64v8/python:3.10.12-slim-buster as build-image
+FROM python:3.10.12-slim-buster as build-image
 
 # Set environment variables
 ENV PYTHONDONTWRITEBYTECODE 1
 ENV PYTHONUNBUFFERED 1
 
 # Set work directory to the root of your project
-WORKDIR /var/task/Model
+WORKDIR var/task/Model
 
 # Install system dependencies
-# RUN apt-get update && apt-get install -y netcat-openbsd
+#RUN apt-get update && apt-get install -y netcat-openbsd
 
 # Install python dependencies
 COPY ./backend/requirements/base.txt ./backend/requirements/base.txt
+# Install and clean up temp caches
 RUN pip install --upgrade pip \
     && pip install -r backend/requirements/base.txt && rm -rf /root/.cache
 
-# Install the AWS Lambda RIC
+# Since we are not using a base AWS image, there is some additional setup required. We need to set up the runtime
+# interface client
+# https://docs.aws.amazon.com/lambda/latest/dg/python-image.html#python-image-clients
+# Additionally install the AWS Lambda RIC
 RUN pip install awslambdaric
 
 # Second stage: "runtime-image"
-FROM arm64v8/python:3.10.12-slim-buster
+FROM python:3.10.12-slim-buster
 
 # Set work directory to the root of your project
 WORKDIR /var/task/Model
@@ -48,4 +48,4 @@ COPY ./utils/ ./utils/
 # Set the ENTRYPOINT to the AWS Lambda RIC and CMD to your function handler
 ENTRYPOINT [ "/usr/local/bin/python", "-m", "awslambdaric" ]
 # Define the handler location
-CMD ["backend.app.main.handler"]
+CMD ["backend.app.main.handler"]
\ No newline at end of file