diff --git a/comfyui/start.sh b/comfyui/start.sh
index 33bfd1b..52b7f7d 100644
--- a/comfyui/start.sh
+++ b/comfyui/start.sh
@@ -16,20 +16,22 @@ export TQDM_DISABLE=1
 # Navigate to ComfyUI directory
 cd "${COMFYUI_DIR}" || exit 1
 
-# Activate virtual environment
-if [ -d "venv" ]; then
-  echo "Activating ComfyUI virtual environment..."
-  source venv/bin/activate
+# Determine which Python to use
+if [ -f "venv/bin/python" ]; then
+  PYTHON_BIN="venv/bin/python"
+  echo "Using ComfyUI virtual environment Python..."
 else
+  PYTHON_BIN="python3"
   echo "WARNING: venv not found, using system Python"
 fi
 
 echo "Starting ComfyUI on port 8188..."
 echo "Access at: http://localhost:8188"
 echo "Using HuggingFace cache: ${HF_CACHE}"
+echo "Python: ${PYTHON_BIN}"
 
 # Start ComfyUI with GPU support
-python main.py \
+exec "${PYTHON_BIN}" main.py \
   --listen 0.0.0.0 \
   --port 8188 \
   --enable-cors-header \