diff --git a/custom_types.yaml b/custom_types.yaml
index 82ab981..c12fcb8 100644
--- a/custom_types.yaml
+++ b/custom_types.yaml
@@ -315,22 +315,32 @@ capability_types:
       version:
         type: list
         required: false
-        entry_schema: version
+        entry_schema:
+          type: version
       publish_all:
         type: boolean
         default: false
         required: false
       publish_ports:
         type: list
-        entry_schema: tosca.datatypes.network.PortSpec
+        entry_schema:
+          type: tosca.datatypes.network.PortSpec
         required: false
       expose_ports:
         type: list
-        entry_schema: tosca.datatypes.network.PortSpec
+        entry_schema:
+          type: tosca.datatypes.network.PortSpec
         required: false
       volumes:
         type: list
-        entry_schema: string
+        entry_schema:
+          type: string
+        required: false
+      host_id:
+        type: string
+        required: false
+      volume_id:
+        type: string
         required: false
 
 artifact_types: