diff --git a/src/test/java/io/cdap/plugin/successfactors/source/metadata/SuccessFactorsSchemaGeneratorTest.java b/src/test/java/io/cdap/plugin/successfactors/source/metadata/SuccessFactorsSchemaGeneratorTest.java
index e131d21..c40cc84 100644
--- a/src/test/java/io/cdap/plugin/successfactors/source/metadata/SuccessFactorsSchemaGeneratorTest.java
+++ b/src/test/java/io/cdap/plugin/successfactors/source/metadata/SuccessFactorsSchemaGeneratorTest.java
@@ -83,7 +83,7 @@ public void testBuildExpandOutputSchema() throws SuccessFactorsServiceException
       "password", "filterOption", "selectOption", "expandOption",
       "paginationType");
     Schema outputSchema = generator.buildExpandOutputSchema("Benefit",
-                                                            "eligibleBenefits", "assEntity", pluginConfig);
+                                                            "eligibleBenefits", "associatedEntity", pluginConfig);
     int lastIndex = outputSchema.getFields().size() - 1;
     Assert.assertEquals("Schema field size is same.",
                         96,