diff --git a/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/GmsGraphQLEngine.java b/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/GmsGraphQLEngine.java
index b83bfca3d7337..59335ba605a74 100644
--- a/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/GmsGraphQLEngine.java
+++ b/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/GmsGraphQLEngine.java
@@ -2383,8 +2383,8 @@ private void configureDataJobResolvers(final RuntimeWiring.Builder builder) {
                             containerType,
                             (env) -> {
                               final DataJob dataJob = env.getSource();
-                              return dataJob.getDataPlatformInstance() != null
-                                  ? dataJob.getDataPlatformInstance().getUrn()
+                              return dataJob.getContainer() != null
+                                  ? dataJob.getContainer().getUrn()
                                   : null;
                             }))
                     .dataFetcher("parentContainers", new ParentContainersResolver(entityClient))