diff --git a/.github/workflows/ml_update.yml b/.github/workflows/ml_update.yml
index 9b6ca05..b9feb39 100644
--- a/.github/workflows/ml_update.yml
+++ b/.github/workflows/ml_update.yml
@@ -28,7 +28,7 @@ jobs:
             echo "${{ secrets.PEM }}" > key.pem 
             set -x
             chmod 600 key.pem
-            scp -i ./"key.pem" ./recommendation_pipeline.sh -o StrictHostKeyChecking=no ec2-user@${{ secrets.CLOUD_URL }}:~
+            scp -i ./"key.pem" -o StrictHostKeyChecking=no ./recommendation_pipeline.sh ec2-user@${{ secrets.CLOUD_URL }}:~
 
         - name: Sync ML Code
           run: |
@@ -36,7 +36,7 @@ jobs:
             echo "${{ secrets.PEM }}" > key.pem 
             set -x
             chmod 600 key.pem
-            rsync -avz ./recommend/ -o StrictHostKeyChecking=no ec2-user@${{ secrets.CLOUD_URL }}:/recommend/
+            rsync -avz -e "ssh -o StrictHostKeyChecking=no -i ./key.pem" ./recommend/ ec2-user@${{ secrets.CLOUD_URL }}:/recommend/
 
         - name: Restart ML Container & Mount Code
           run: |
diff --git a/recommend/entrypoint.sh b/recommend/entrypoint.sh
index b2df974..c2f1029 100644
--- a/recommend/entrypoint.sh
+++ b/recommend/entrypoint.sh
@@ -6,9 +6,9 @@
 set -e
 
 # Set environment variables for S3 bucket and file paths
-# S3_BUCKET=${S3_BUCKET}
-# EVENT_FILE="eventData.csv"
-# USER_FILE="userData.csv"
+S3_BUCKET=${S3_BUCKET}
+EVENT_FILE="eventData.csv"
+USER_FILE="userData.csv"
 
 # Use AWS CLI to download the CSV file from S3
 aws s3 cp s3://${S3_BUCKET}/${EVENT_FILE} /app/