diff --git a/main.tf b/main.tf
index f77ca07..c70ef49 100644
--- a/main.tf
+++ b/main.tf
@@ -45,6 +45,7 @@ resource "castai_node_configuration" "this" {
   aks {
     max_pods_per_node = try(each.value.max_pods_per_node, 30)
     os_disk_type      = try(each.value.os_disk_type, null)
+    aks_image_family  = try(each.value.aks_image_family, null)
   }
 }