diff --git a/scripts/deploy.sh b/scripts/deploy.sh
index 0dcb65d..10df645 100755
--- a/scripts/deploy.sh
+++ b/scripts/deploy.sh
@@ -19,7 +19,7 @@ VOLUME_SIZE=${VOLUME_SIZE:-200}                     # GB
 VOLUME_PATH=${VOLUME_PATH:-"/root/data"}            # Mount path within container
 
 # Docker configuration
-DOCKER_IMAGE=${DOCKER_IMAGE:-"runpod/pytorch:2.1.0-py3.10-cuda11.8.0-devel-ubuntu22.04"}
+DOCKER_IMAGE=${DOCKER_IMAGE:-"runpod/vscode-server:0.0.0"}
 
 # Load environment variables from .env file if it exists
 if [ -f .env ]; then