diff --git a/modules/vmware/main.tf b/modules/vmware/main.tf
index 3552257..4ebc601 100644
--- a/modules/vmware/main.tf
+++ b/modules/vmware/main.tf
@@ -5,8 +5,7 @@ variable "vms" {
         cpu = number
         ram = number
         guest_id = string
-        disks = list(object({
-          name = string
+        disks = map(object({
           size = string
         }))
         network_interfaces = map(object(
@@ -77,7 +76,7 @@ resource "vphere_virtual_virtual_machine" "vm" {
   dynamic "disk" {
     for_each = each.key.disks 
     content {
-      label = each.key.name
+      label = each.key
       size = each.key.size
     }
   }