diff --git a/setup-runpod.sh b/setup-runpod.sh
index fa26097ab..35ca3d5eb 100644
--- a/setup-runpod.sh
+++ b/setup-runpod.sh
@@ -1,3 +1,5 @@
+#!/usr/bin/env bash
+
 # This gets the directory the script is run from so pathing can work relative to the script where needed.
 SCRIPT_DIR="$(cd -- $(dirname -- "$0") && pwd)"
 
@@ -7,8 +9,13 @@ if [ ! -d "$SCRIPT_DIR/venv" ]; then
     python3 -m venv venv
 fi
 
+# Install tk
+echo "Install tk..."
+apt update -y && apt install -y python3-tk
+
 # Activate the virtual environment
-source "$SCRIPT_DIR/venv/bin/activate"
+echo "Activate venv..."
+source "$SCRIPT_DIR/venv/bin/activate" || exit 1
 
 # Run setup_linux.py script with platform requirements
 python "$SCRIPT_DIR/setup/setup_linux.py" --platform-requirements-file=requirements_runpod.txt