diff --git a/documentation/IDTA-01001/modules/ROOT/partials/diagrams/90-submodel-element.puml b/documentation/IDTA-01001/modules/ROOT/partials/diagrams/90-submodel-element.puml
index e20970f5c..a3400d3c4 100644
--- a/documentation/IDTA-01001/modules/ROOT/partials/diagrams/90-submodel-element.puml
+++ b/documentation/IDTA-01001/modules/ROOT/partials/diagrams/90-submodel-element.puml
@@ -4,15 +4,20 @@
 
 !include https://raw.githubusercontent.com/admin-shell-io/aas-specs/IDTA-01001-3-1_working/documentation/IDTA-01001/modules/ROOT/partials/diagrams/classes/submodel-element.puml
 abstract class SubmodelElement {
+  .. ::Referable ..
   +idShort: IdShortType[0..1]
   +displayName: MultiLanguageNameType[0..1]
   +description: MultiLanguageTextType[0..1]
+  +category: NameType[0..1] <<Deprecated>>
+    .. ::HasExtensions ..
   +extension: Extension[0..*]
+      .. ::HasSemantics ..
   +semanticId: Reference[0..1]
   +supplementalSemanticId: Reference[0..*]
+        .. ::Qualifiable ..
   +qualifier: Qualifier[0..*]
-  +embeddedDataSpecification: EmbeddedDataSpecification[0..*]
-  +category: NameType[0..1] <<Deprecated>>
+        .. ::HasDataSpecification ..
+  +dataSpecification: Reference[0..*]
 }
 
 !include https://raw.githubusercontent.com/admin-shell-io/aas-specs/IDTA-01001-3-1_working/documentation/IDTA-01001/modules/ROOT/partials/diagrams/classes/relationship-element.puml