diff --git a/backend/docker/Dockerfile b/backend/docker/Dockerfile
index 24f319db..a02fda22 100644
--- a/backend/docker/Dockerfile
+++ b/backend/docker/Dockerfile
@@ -6,7 +6,7 @@ ENV PYTHONDONTWRITEBYTECODE 1
 ENV PYTHONUNBUFFERED 1
 
 # Set work directory to the root of your project
-WORKDIR /Model
+WORKDIR var/task/Model
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y netcat-openbsd
diff --git a/backend/docker/lambda.Dockerfile b/backend/docker/lambda.Dockerfile
index dab7275c..a2a1de99 100644
--- a/backend/docker/lambda.Dockerfile
+++ b/backend/docker/lambda.Dockerfile
@@ -6,7 +6,7 @@ ENV PYTHONDONTWRITEBYTECODE 1
 ENV PYTHONUNBUFFERED 1
 
 # Set work directory to the root of your project
-WORKDIR /Model
+WORKDIR var/task/Model
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y netcat-openbsd
@@ -31,4 +31,4 @@ COPY ./datatypes/ ./datatypes/
 COPY ./utils/ ./utils/
 
 # Define the handler location
-#CMD ["backend.app.main.handler"]
+CMD ["backend.app.main.handler"]
diff --git a/serverless.yml b/serverless.yml
index 3445e0f9..89eb666a 100644
--- a/serverless.yml
+++ b/serverless.yml
@@ -43,7 +43,6 @@ custom:
 
 functions:
   app:
-    handler: backend.app.main.handler
     image:
       uri: ${env:ECR_URI}:${env:GITHUB_SHA}
     events: