diff --git a/dao-impl/ebean-dao/src/main/java/com/linkedin/metadata/dao/EbeanLocalRelationshipQueryDAO.java b/dao-impl/ebean-dao/src/main/java/com/linkedin/metadata/dao/EbeanLocalRelationshipQueryDAO.java
index b60d57bf6..2f88d8449 100644
--- a/dao-impl/ebean-dao/src/main/java/com/linkedin/metadata/dao/EbeanLocalRelationshipQueryDAO.java
+++ b/dao-impl/ebean-dao/src/main/java/com/linkedin/metadata/dao/EbeanLocalRelationshipQueryDAO.java
@@ -260,6 +260,7 @@ protected boolean isMgEntityType(@Nonnull String entityType) {
     return _mgEntityTypeNameSet.contains(StringUtils.lowerCase(entityType));
   }
 
+
   /**
    * Extracts the table name from an entity urn for MG entities. If entityUrn is not for MG entity, return null.
    * @param entityType String representing the type of entity (e.g. "dataset")
@@ -481,7 +482,7 @@ private String buildFindRelationshipSQL(
   /**
    * Creates a set of MG entity type names by querying the database.
    */
-  private void initMgEntityTypeNameSet() {
+  public void initMgEntityTypeNameSet() {
     final String sql = "SELECT table_name FROM information_schema.tables"
         + " WHERE table_type = 'BASE TABLE' AND TABLE_SCHEMA=DATABASE() AND table_name LIKE 'metadata_entity_%'";
     _mgEntityTypeNameSet = _server.createSqlQuery(sql).findList().stream()