diff --git a/pkgs/by-name/az/azure-cli/extensions-generated.json b/pkgs/by-name/az/azure-cli/extensions-generated.json
index 1a938c126902..9aa7b5a2d69b 100644
--- a/pkgs/by-name/az/azure-cli/extensions-generated.json
+++ b/pkgs/by-name/az/azure-cli/extensions-generated.json
@@ -1147,13 +1147,6 @@
     "hash": "sha256-kIsN8HzvZSF2oPK/D9z1i10W+0kD7jwG9z8Ls5E6XA8=",
     "description": "Additional commands for Azure AppService"
   },
-  "workload-orchestration": {
-    "pname": "workload-orchestration",
-    "version": "5.1.0",
-    "url": "https://azcliprod.blob.core.windows.net/cli-extensions/workload_orchestration-5.1.0-py3-none-any.whl",
-    "hash": "sha256-/9MGhiqQpEUPJ9AgVM0tZ/SQiRdVvOsXVgdnqAtfSMw=",
-    "description": "Microsoft Azure Command-Line Tools WorkloadOperations Extension"
-  },
   "workloads": {
     "pname": "workloads",
     "version": "1.1.0",
diff --git a/pkgs/by-name/az/azure-cli/extensions-manual.nix b/pkgs/by-name/az/azure-cli/extensions-manual.nix
index 8d361268f807..0774e4acc239 100644
--- a/pkgs/by-name/az/azure-cli/extensions-manual.nix
+++ b/pkgs/by-name/az/azure-cli/extensions-manual.nix
@@ -359,6 +359,16 @@
     ];
     meta.maintainers = with lib.maintainers; [ techknowlogick ];
   };
+  workload-orchestration = mkAzExtension {
+    pname = "workload-orchestration";
+    version = "5.1.1";
+    url = "https://azcliprod.blob.core.windows.net/cli-extensions/workload_orchestration-5.1.1-py3-none-any.whl";
+    hash = "sha256-mtmFRd6K2uOpzgKezdAoBDD7mGFh7blkUGvMqSajdSQ=";
+    description = "Microsoft Azure Command-Line Tools WorkloadOperations Extension";
+    propagatedBuildInputs = with python3Packages; [
+      kubernetes
+    ];
+  };
 }
 // lib.optionalAttrs config.allowAliases {
   # Removed extensions