Update EntityExtensions.cs