diff --git a/datahub-web-react/src/graphql/fragments.graphql b/datahub-web-react/src/graphql/fragments.graphql
index 513857b8b4f3c..0b1e833ca2ee8 100644
--- a/datahub-web-react/src/graphql/fragments.graphql
+++ b/datahub-web-react/src/graphql/fragments.graphql
@@ -1148,6 +1148,10 @@ fragment nonRecursiveMLModelGroupFields on MLModelGroup {
             trainingJobs
             downstreamJobs
         }
+        customProperties {
+            key
+            value
+        }
     }
     browsePathV2 {
         ...browsePathV2Fields