diff --git a/mkosi.extra/usr/bin/set-up-systemd-extension b/mkosi.extra/usr/bin/set-up-systemd-extension
index b1330d0..6aa3dec 100644
--- a/mkosi.extra/usr/bin/set-up-systemd-extension
+++ b/mkosi.extra/usr/bin/set-up-systemd-extension
@@ -7,11 +7,7 @@
 set -e
 
 # Test to see if the user added a name for their extension. Otherwise use 'kde'
-SYSEXT_NAME="kde"
-
-if test -n "$1"; then
-  SYSEXT_NAME="$1"
-fi
+SYSEXT_NAME=${1:-"kde"}
 
 SYSEXT_BASE_LOCATION="${HOME}/${SYSEXT_NAME}/usr/"
 SYSEXT_LOCATION="${SYSEXT_BASE_LOCATION}/lib/extension-release.d/"