diff --git a/cmd/project/functions.go b/cmd/project/functions.go
index 881d962..b976bf1 100644
--- a/cmd/project/functions.go
+++ b/cmd/project/functions.go
@@ -353,7 +353,11 @@ func startProject(networkVolumeId string) error {
 	handlerPath := path.Join(remoteProjectPath, config.GetPath([]string{"runtime", "handler_path"}).(string))
 	launchApiServer := fmt.Sprintf(`
 		#!/bin/bash
-		API_PORT=7271
+		if [ -z "${BASE_RELEASE_VERSION}" ]; then
+			API_PORT=7270
+		else
+			API_PORT=7271
+		fi
 		API_HOST="0.0.0.0"
 		PYTHON_VENV_PATH="%s" # Path to the Python virutal environment used during development located on the Pod at /<project_id>/venv
 		PROJECT_DIRECTORY="%s/%s"
diff --git a/cmd/project/tomlBuilder.go b/cmd/project/tomlBuilder.go
index 9131217..752b12a 100644
--- a/cmd/project/tomlBuilder.go
+++ b/cmd/project/tomlBuilder.go
@@ -32,7 +32,7 @@ name = "%s"
 # container_disk_size_gb - Disk space allocated to the container. Adjust according to your needs.
 
 uuid = "%s"
-base_image = "runpod/base:0.5.2-cuda%s"
+base_image = "runpod/base:0.6.1-cuda%s"
 gpu_types = [
     "NVIDIA GeForce RTX 4080",  # 16GB
     "NVIDIA RTX A4000",         # 16GB