diff --git a/deployment/serverless.yml b/deployment/serverless.yml
index b9f4ac4..009f7db 100644
--- a/deployment/serverless.yml
+++ b/deployment/serverless.yml
@@ -4,7 +4,6 @@ provider:
   name: aws
   region: eu-west-2
   architecture: x86_64
-  memorySize: 4096
   environment:
     RUNTIME_ENVIRONMENT: ${env:RUNTIME_ENVIRONMENT}
     PREDICTIONS_BUCKET: ${env:PREDICTIONS_BUCKET}
@@ -52,3 +51,4 @@ functions:
           path: /predict
           method: POST
     timeout: 120 # Set max run time to 2 minutes - we shouldn't need this much time so this can be reviewed
+    memorySize: 4096