diff --git a/.github/workflows/deploy.yml b/.github/workflows/deploy.yml
index d49e531..912a67a 100644
--- a/.github/workflows/deploy.yml
+++ b/.github/workflows/deploy.yml
@@ -16,3 +16,4 @@ jobs:
       aws-key-id: ${{ secrets.LAMBDA_CONTAINER_PIPELINE_AWS_ID }}
       aws-secret-key: ${{ secrets.LAMBDA_CONTAINER_PIPELINE_AWS_SECRET}}
       function-admin-api-key: ${{ secrets.FUNCTION_ADMIN_API_KEY}}
+      license: ${{ secrets.LICENSE }}
diff --git a/Dockerfile b/Dockerfile
index a4aedad..c073218 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM ghcr.io/lambda-feedback/evaluation-function-base/wolfram:latest
+FROM ghcr.io/lambda-feedback/evaluation-function-base/wolfram:latest as base
 
 # Command to start the evaluation function with
 ENV FUNCTION_COMMAND="wolframscript"
@@ -11,8 +11,5 @@ ENV FUNCTION_INTERFACE="file"
 
 ENV LOG_LEVEL="DEBUG"
 
-# Copy Wolfram licence if present - stored as LICENCE.txt for working with the lambda_build Github Actions workflow
-RUN if [ -f dist/LICENSE.txt ]; then cp dist/LICENSE.txt /home/wolframengine/.WolframEngine/Licensing/mathpass; else echo "No license file found."; fi
-
 # Copy the evaluation function to the app directory
 COPY ./evaluation_function.wl /app/evaluation_function.wl
\ No newline at end of file