diff --git a/nixos/modules/update-deployment b/nixos/modules/update-deployment
index 43f1ad918231442b68fff2dc62ade9b858ddffcb..82bbc2129e83743b92f599c9517d49021bd7d0e3 100755
--- a/nixos/modules/update-deployment
+++ b/nixos/modules/update-deployment
@@ -6,11 +6,23 @@ set -euxo pipefail
 GRIDNAME=$1
 shift
 
-if [ "${GRIDNAME}" = "local" ]; then
-    BRANCH="323.continuous-deployment"
-else
-    BRANCH="${GRIDNAME}"
-fi
+case "${GRIDNAME}" in
+    "local")
+	BRANCH="323.continuous-deployment"
+	;;
+
+    "testing")
+	BRANCH="staging"
+	;;
+
+    "production")
+	BRANCH="production"
+	;;
+
+    *)
+	echo "Unknown grid: ${GRIDNAME}"
+	exit 1
+esac
 
 CHECKOUT="${HOME}/PrivateStorageio"
 REPO="https://whetstone.privatestorage.io/privatestorage/PrivateStorageio.git"