diff --git a/datahub-web-react/src/graphql/fragments.graphql b/datahub-web-react/src/graphql/fragments.graphql
index 56b5dc86254097..344b4db26027a0 100644
--- a/datahub-web-react/src/graphql/fragments.graphql
+++ b/datahub-web-react/src/graphql/fragments.graphql
@@ -897,6 +897,10 @@ fragment nonRecursiveMLModel on MLModel {
             key
             value
         }
+        mlModelLineageInfo {
+            trainingJobs
+            downstreamJobs
+        }
     }
     globalTags {
         ...globalTagsFields
@@ -979,6 +983,10 @@ fragment nonRecursiveMLModelGroupFields on MLModelGroup {
     browsePathV2 {
         ...browsePathV2Fields
     }
+    mlModelLineageInfo {
+        trainingJobs
+        downstreamJobs
+    }
 }
 
 fragment platformFields on DataPlatform {