From 21e00bc16ce54999501ce804758f0bd4b0b2931c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sergio=20G=C3=B3mez=20Villamor?= <sgomezvillamor@gmail.com>
Date: Thu, 9 Jan 2025 09:38:29 +0100
Subject: [PATCH] fix datajob resolver

---
 .../java/com/linkedin/datahub/graphql/GmsGraphQLEngine.java   | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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 b83bfca3d7337e..59335ba605a741 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))