diff --git a/.github/workflows/Deploy.yml b/.github/workflows/Deploy.yml
index 330855c..5f42eae 100644
--- a/.github/workflows/Deploy.yml
+++ b/.github/workflows/Deploy.yml
@@ -22,6 +22,10 @@ jobs:
           npm install -g serverless
           npm install -g serverless-domain-manager
 
+      - name: Install DVC
+        run: |
+          pip install dvc
+
       # Set up all of the secrets required for the deployment
       - name: set secret prefix which is used across multiple steps
         id: secret_prefix
@@ -79,6 +83,11 @@ jobs:
           aws-secret-access-key: ${{ secrets.PROD_AWS_SECRET_ACCESS_KEY }}
           aws-region: eu-west-2
 
+      - name: DVC Pull
+        run: |
+          cd modules/ml-pipeline/src/pipeline
+          dvc pull -r ${{ steps.set_runtime_environment.outputs.runtime_environment }}
+
       - name: Setup Docker
         uses: docker/setup-buildx-action@v1
 
diff --git a/deployment/Dockerfile.prediction.lambda b/deployment/Dockerfile.prediction.lambda
index b4f2323..58c3a88 100644
--- a/deployment/Dockerfile.prediction.lambda
+++ b/deployment/Dockerfile.prediction.lambda
@@ -19,7 +19,6 @@ RUN pip install --no-cache-dir -r ./requirements.txt
 COPY modules/ml-pipeline/src/pipeline ./pipeline
 # Copy the handler
 COPY deployment/handlers/prediction_app.py prediction_app.py
-# Get the model
-# RUN dvc pull -r ${RUNTIME_ENVIRONMENT}
+
 
 CMD [ "prediction_app.handler" ]