diff --git a/modules/gluster-module/disks.tf b/modules/gluster-module/disks.tf
index b6863e84..b006d0d2 100644
--- a/modules/gluster-module/disks.tf
+++ b/modules/gluster-module/disks.tf
@@ -12,7 +12,7 @@ resource "nebius_compute_v1_disk" "glusterfs-boot-disk" {
   block_size_bytes    = 4096
   size_bytes          = 64424509440
   type                = "NETWORK_SSD"
-  source_image_family = { image_family = "ubuntu22.04-cuda12" }
+  source_image_family = "ubuntu22.04-cuda12"
 }
 
 resource "nebius_compute_v1_disk" "glusterfs-storage-disk" {
@@ -22,4 +22,4 @@ resource "nebius_compute_v1_disk" "glusterfs-storage-disk" {
   block_size_bytes = var.disk_block_size
   size_bytes       = var.disk_size # 100 GB
   type             = var.disk_type
-}
+}
\ No newline at end of file