diff --git a/.github/workflows/on_push_deploy_to_staging.yaml b/.github/workflows/on_push_deploy_to_staging.yaml
index 22a019e30..00397258f 100644
--- a/.github/workflows/on_push_deploy_to_staging.yaml
+++ b/.github/workflows/on_push_deploy_to_staging.yaml
@@ -6,12 +6,8 @@ on:
       - master
     paths:
       - ".github/workflows/on_push_deploy_to_staging.yaml"
-<<<<<<< Updated upstream
-      # TODO: Decide if we want to add other file changes here. 
-      # TODO: Decide how we handle updating this forked copy.
-=======
       # TODO: Decide if we want to add other file changes here.
->>>>>>> Stashed changes
+      # TODO: Decide how we handle updating this forked copy.
 jobs:
   deploy:
     runs-on: ubuntu-latest
diff --git a/.github/workflows/on_tag_deploy_to_production.yaml b/.github/workflows/on_tag_deploy_to_production.yaml
index 2c724834e..1fd4cd20e 100644
--- a/.github/workflows/on_tag_deploy_to_production.yaml
+++ b/.github/workflows/on_tag_deploy_to_production.yaml
@@ -74,6 +74,8 @@ jobs:
             
             # Run the new Docker container
             docker run --gpus all -d -p 7861:7861 --name moonlite-sd \
+              -v /home/ec2-user/apps/stable-diffusion-webui/outputs/img2img-images:/outputs/img2img-images \
+              -v /home/ec2-user/apps/media_root:/home/ec2-user/apps/media_root \
               --env-file /home/${{ secrets.PROD_EC2_USER }}/.env \
               ${{ secrets.PROD_ECR_URI }}:latest