diff --git a/k8s-training/main.tf b/k8s-training/main.tf
index fa140d14..e877161f 100644
--- a/k8s-training/main.tf
+++ b/k8s-training/main.tf
@@ -69,7 +69,7 @@ resource "nebius_mk8s_v1_node_group" "gpu" {
     network_interfaces = [
       {
         subnet_id = var.subnet_id
-        public_ip = var.gpu_nodes_assign_public_ip ? {} : null
+        public_ip_address = var.gpu_nodes_assign_public_ip ? {} : null
       }
     ]
     resources = {