diff --git a/augments/configuration.yang b/augments/configuration.yang
index 1ee8273..94a226d 100644
--- a/augments/configuration.yang
+++ b/augments/configuration.yang
@@ -54,6 +54,7 @@
     grouping extended-kdu-configuration {
         list kdu-profile {
             key "name";
+
             leaf name {
                 description
                   "Reference to the KDU";
@@ -61,6 +62,7 @@
                     path "/vnfd:vnfd/kdu:kdu/kdu:name";
                 }
             }
+
             leaf kdu-configuration-id {
                 description
                   "Reference to the KDU configuration to use in this DF";
@@ -68,6 +70,14 @@
                     path "/vnfd:vnfd/kdu:kdu-configuration/kdu:id";
                 }
             }
+
+            leaf kdu-model-id {
+                description
+                  "Reference to the KDU model to use in this DF";
+                type leafref {
+                    path "/vnfd:vnfd/kdu:kdu-model/kdu:id";
+                }
+            }
         }
     }
 
diff --git a/augments/kdu.yang b/augments/kdu.yang
index edc93f4..0cad701 100644
--- a/augments/kdu.yang
+++ b/augments/kdu.yang
@@ -64,6 +64,36 @@
         }
     }
 
+    grouping extended-kdu-model {
+        list kdu-model {
+            key "id";
+
+            leaf id {
+                description
+                  "Internal identifier for the KDU model";
+                type string;
+            }
+
+            leaf kdu-model-type {
+                description
+                  "Indicates the KDU model, either as a helm-chart or as a juju-bundle.";
+
+                type enumeration {
+                    enum helm-chart;
+                    enum juju-bundle;
+                }
+            }
+
+            leaf kdu-model-locator {
+                description
+                  "Indicates the KDU model location, either as a path to a folder in the 
+                   package or as a URL where to fetch the model.";
+
+                type string;
+            }
+        }
+    }
+
     grouping extended-kdu {
         list kdu {
             description
@@ -81,26 +111,6 @@
                   "Description of the KDU.";
                 type string;
             }
-
-            container kdu-model {
-                leaf kdu-model-type {
-                    description
-                      "Indicates the KDU model, either as a helm-chart or as a juju-bundle.";
-
-                    type enumeration {
-                        enum helm-chart;
-                        enum juju-bundle;
-                    }
-                }
-
-                leaf kdu-model-locator {
-                    description
-                      "Indicates the KDU model location, either as a path to a folder in the 
-                       package or as a URL where to fetch the model.";
-
-                    type string;
-                }
-            }
         }
 
         container k8s-cluster {
@@ -146,6 +156,7 @@
     augment "/vnfd:vnfd" {
         uses extended-kdu;
         uses extended-kdu-configuration;
+        uses extended-kdu-model;
     }
 
     augment "/vnfd:vnfd/vnfd:ext-cpd/vnfd:cp-connection" {