diff --git a/dietpi/build.sh b/dietpi/build.sh
index 0650d2a..3c4b2e7 100755
--- a/dietpi/build.sh
+++ b/dietpi/build.sh
@@ -6,8 +6,12 @@
 HERE="$(dirname "$(realpath "$0")")"
 REPO_DIR=$(realpath "$HERE/..")
 
+# Add /boot/dietpi to PATH so we can use the dietpi-* commands
+PATH="$PATH:/boot/dietpi"
 source "$HERE/utils.sh"
 
+#--------------------------------------------------------------------------------------------------
+
 ## Create pi-dj user and delete default dietpi user
 default_username=pi-dj
 default_password=pidj
@@ -56,6 +60,7 @@ set_dietpi_config SOFTWARE_DISABLE_SSH_PASSWORD_LOGINS root # Disable SSH passwo
 #
 #
 
+
 ## Install all packages needed for a usable desktop environment
 # 5: Alsa
 # 6: X.Org
@@ -65,11 +70,12 @@ set_dietpi_config SOFTWARE_DISABLE_SSH_PASSWORD_LOGINS root # Disable SSH passwo
 # 188: Go 
 # 67: Firefox
 dietpi-software install 5 6 7 170 17 188 67
+
 # install i3 and dependencies
-apt install -y i3 polybar onboard fonts-roboto fonts-font-awesome dunst rofi
+apt-get install -y i3 polybar onboard fonts-roboto fonts-font-awesome dunst rofi
 
 # install some useful CLI tools for maintenance
-apt install -y kitty zsh man less
+apt-get install -y kitty zsh man less
 sh -c "$(curl -fsSL https://raw.githubusercontent.com/ohmyzsh/ohmyzsh/master/tools/install.sh)"
 chsh -s "$(which zsh)"
 git config pull.rebase true