diff --git a/nixos-rebuild.sh b/nixos-rebuild.sh
index 5905b54..ad28160 100755
--- a/nixos-rebuild.sh
+++ b/nixos-rebuild.sh
@@ -36,7 +36,7 @@ EXPECTED_RESPONSE=$(cat << EOF
   "data": {
     "__typename": "Mutation",
     "system": {
-      "runSystemRebuild": {
+      "runSystem${REBUILD_OP}": {
         "code": 200,
         "success": true
       }