diff --git a/deploy/aws_ami/docker-compose.yml b/deploy/aws_ami/docker-compose.yml
index a7a4f49429..7479e643c0 100644
--- a/deploy/aws_ami/docker-compose.yml
+++ b/deploy/aws_ami/docker-compose.yml
@@ -2,7 +2,7 @@ version: "3"
 
 services:
   appsmith:
-    image: index.docker.io/appsmith/appsmith-ce
+    image: index.docker.io/appsmith/appsmith-ee
     container_name: appsmith
     ports:
       - "80:80"
@@ -26,4 +26,4 @@ services:
   #  depends_on:
   #    - appsmith
   #  environment:
-  #    - WATCHTOWER_LIFECYCLE_HOOKS=true
\ No newline at end of file
+  #    - WATCHTOWER_LIFECYCLE_HOOKS=true
diff --git a/deploy/digital_ocean/files/var/lib/cloud/scripts/per-instance/01-onboot b/deploy/digital_ocean/files/var/lib/cloud/scripts/per-instance/01-onboot
index 24790153f0..25e0056f59 100755
--- a/deploy/digital_ocean/files/var/lib/cloud/scripts/per-instance/01-onboot
+++ b/deploy/digital_ocean/files/var/lib/cloud/scripts/per-instance/01-onboot
@@ -9,7 +9,7 @@ cd /root/appsmith
 
 # Step 1: Download the templates
 echo "Downloading the Docker Compose file..."
-curl -L https://bit.ly/3AQzII6 -o $PWD/docker-compose.yml
+curl -L https://raw.githubusercontent.com/appsmithorg/appsmith/master/deploy/aws_ami/docker-compose.yml -o $PWD/docker-compose.yml
 
 # Step 2: Pulling the latest container images
 echo ""