diff --git a/server/scripts/deploy.sh b/server/scripts/deploy.sh
index ba41e73..fd96c4f 100644
--- a/server/scripts/deploy.sh
+++ b/server/scripts/deploy.sh
@@ -58,13 +58,13 @@ switch_container() {
     echo "Starting $next_container..."
     docker compose -f compose.yml up "$next_container" -d
 
-    # sleep 5
+    sleep 5
 
-    # if ! health_check "$health_check_url"; then
-    #     echo "Health check failed, rolling back..."
-    #     docker compose -f compose.yml down "$next_container"
-    #     return
-    # fi
+    if ! health_check "$health_check_url"; then
+        echo "Health check failed, rolling back..."
+        docker compose -f compose.yml down "$next_container"
+        return
+    fi
 
     sed -i "s/server $prev_container:8080;/server $next_container:8080;/" "$NGINX_CONFIG_PATH"
 
@@ -75,6 +75,8 @@ switch_container() {
     fi
 
     echo "$next_container is now live and accepting requests!"
+
+    docker compose -f compose.yml down "$prev_container"
 }
 
 IS_GREEN=$(docker container ps | grep app-green)