Fixed bug with updating records with EF