diff --git a/bootstrap.sh b/bootstrap.sh
index 6cc3e32..007b070 100644
--- a/bootstrap.sh
+++ b/bootstrap.sh
@@ -17,7 +17,13 @@ if [ -z "$USERNAME" ]; then
 fi
 
 SSH_KEY_URL="https://skulldev.de/Skull-IT/trusted-ssh-keys/raw/branch/main/trusted-ssh-keys"
-SSH_DIR="/home/$USERNAME/.ssh"
+
+# Set SSH directory based on username
+if [ "$USERNAME" = "root" ]; then
+    SSH_DIR="/root/.ssh"
+else
+    SSH_DIR="/home/$USERNAME/.ssh"
+fi
 
 # Update package list and install necessary packages
 echo "[+] Updating package list and installing essential packages..."