diff --git a/dao-api/src/main/java/com/linkedin/metadata/dao/utils/GraphUtils.java b/dao-api/src/main/java/com/linkedin/metadata/dao/utils/GraphUtils.java
index 493dcdf17..8f47d3fa6 100644
--- a/dao-api/src/main/java/com/linkedin/metadata/dao/utils/GraphUtils.java
+++ b/dao-api/src/main/java/com/linkedin/metadata/dao/utils/GraphUtils.java
@@ -25,7 +25,7 @@ public static void checkSameSourceUrn(@Nonnull final List<? extends RecordTempla
     }
 
     // TODO: remove this temporary allow list after AIM no longer relies on ProducesLocalRelationshipBuilderFromFeatureDependencies
-    if (assetUrn != null && assetUrn.getEntityType().equals("mlFeatureVersion") && relationships.get(0).getClass().getSimpleName().equals("produces")) {
+    if (assetUrn != null && assetUrn.getEntityType().equals("mlFeatureVersion") && relationships.get(0).getClass().getSimpleName().equals("Produces")) {
       return;
     }