diff --git a/src/Umbraco.Core/UdiEntityTypeHelper.cs b/src/Umbraco.Core/UdiEntityTypeHelper.cs
index c6f884db16cc..a9259521d40d 100644
--- a/src/Umbraco.Core/UdiEntityTypeHelper.cs
+++ b/src/Umbraco.Core/UdiEntityTypeHelper.cs
@@ -111,6 +111,10 @@ public static UmbracoObjectTypes ToUmbracoObjectType(string entityType)
                 return UmbracoObjectTypes.FormsDataSource;
             case Constants.UdiEntityType.Language:
                 return UmbracoObjectTypes.Language;
+            case Constants.UdiEntityType.Element:
+                return UmbracoObjectTypes.Element;
+            case Constants.UdiEntityType.ElementContainer:
+                return UmbracoObjectTypes.ElementContainer;
         }
 
         throw new NotSupportedException(