diff --git a/internal/services/containers/kubernetes_cluster_node_pool_resource.go b/internal/services/containers/kubernetes_cluster_node_pool_resource.go
index 19af0b55d4fb..d9b85e7dec98 100644
--- a/internal/services/containers/kubernetes_cluster_node_pool_resource.go
+++ b/internal/services/containers/kubernetes_cluster_node_pool_resource.go
@@ -1172,7 +1172,6 @@ func upgradeSettingsSchema() *pluginsdk.Schema {
 				"drain_timeout_in_minutes": {
 					Type:     pluginsdk.TypeInt,
 					Optional: true,
-					Default:  30,
 				},
 				"node_soak_duration_in_minutes": {
 					Type:         pluginsdk.TypeInt,
diff --git a/internal/services/containers/kubernetes_cluster_node_pool_resource_test.go b/internal/services/containers/kubernetes_cluster_node_pool_resource_test.go
index 25a2d11c7a7a..0cdb6b7eb58b 100644
--- a/internal/services/containers/kubernetes_cluster_node_pool_resource_test.go
+++ b/internal/services/containers/kubernetes_cluster_node_pool_resource_test.go
@@ -912,13 +912,6 @@ func TestAccKubernetesClusterNodePool_workloadRuntime(t *testing.T) {
 			),
 		},
 		data.ImportStep(),
-		{
-			Config: r.workloadRuntime(data, "WasmWasi"),
-			Check: acceptance.ComposeTestCheckFunc(
-				check.That(data.ResourceName).ExistsInAzure(r),
-			),
-		},
-		data.ImportStep(),
 		{
 			Config: r.workloadRuntime(data, "KataMshvVmIsolation"),
 			Check: acceptance.ComposeTestCheckFunc(
@@ -2365,7 +2358,7 @@ resource "azurerm_kubernetes_cluster_node_pool" "test" {
   vm_size               = "Standard_DS2_v2"
   enable_auto_scaling   = true
   min_count             = 1
-  max_count             = 1000
+  max_count             = 399
   node_count            = 1
 }
 `, r.templateConfig(data))
diff --git a/internal/services/containers/kubernetes_cluster_other_resource_test.go b/internal/services/containers/kubernetes_cluster_other_resource_test.go
index ccb36bf9b856..ed530f70ef35 100644
--- a/internal/services/containers/kubernetes_cluster_other_resource_test.go
+++ b/internal/services/containers/kubernetes_cluster_other_resource_test.go
@@ -1358,7 +1358,7 @@ resource "azurerm_kubernetes_cluster" "test" {
     vm_size             = "Standard_DS2_v2"
     enable_auto_scaling = true
     min_count           = 1
-    max_count           = 1000
+    max_count           = 399
     node_count          = 1
     upgrade_settings {
       max_surge = "10%%"