diff --git a/metadata-models/src/main/resources/entity-registry.yml b/metadata-models/src/main/resources/entity-registry.yml
index 87dc120a788e5..cf33d97426ecf 100644
--- a/metadata-models/src/main/resources/entity-registry.yml
+++ b/metadata-models/src/main/resources/entity-registry.yml
@@ -377,6 +377,7 @@ entities:
       - testResults
       - versionProperties
       - subTypes
+      - container
   - name: mlModelGroup
     category: core
     keyAspect: mlModelGroupKey
@@ -396,6 +397,7 @@ entities:
       - forms
       - testResults
       - subTypes
+      - container
   - name: mlModelDeployment
     category: core
     keyAspect: mlModelDeploymentKey
@@ -407,6 +409,7 @@ entities:
       - globalTags
       - dataPlatformInstance
       - testResults
+      - container
   - name: mlFeatureTable
     category: core
     keyAspect: mlFeatureTableKey