diff --git a/src/OpenRiaServices.Tools/Test/CodeGenExternalAttributeTests.cs b/src/OpenRiaServices.Tools/Test/CodeGenExternalAttributeTests.cs
index d272941d5..c2ec0a2d9 100644
--- a/src/OpenRiaServices.Tools/Test/CodeGenExternalAttributeTests.cs
+++ b/src/OpenRiaServices.Tools/Test/CodeGenExternalAttributeTests.cs
@@ -54,7 +54,7 @@ public void CodeGen_External_Entity_EFtoPOCO()
             TestHelper.AssertGeneratedCodeContains(
                 generatedCode,
                 "private EntityRef<global::DataTests.Scenarios.EF.Northwind.PersonalDetails> _personalDetails_MarkedAsExternal;",
-                "[Association(\"Employee_PersonalDetails\", \"EmployeeID\", \"UniqueID\", IsForeignKey=true)] [ExternalReference()]",
+                "[EntityAssociation(\"Employee_PersonalDetails\", new string[] { \"EmployeeID\"}, new string[] { \"UniqueID\"}, IsForeignKey=true)] [ExternalReference()]",
                 "public global::DataTests.Scenarios.EF.Northwind.PersonalDetails PersonalDetails_MarkedAsExternal",
                 "private bool FilterPersonalDetails_MarkedAsExternal(global::DataTests.Scenarios.EF.Northwind.PersonalDetails entity)");
         }