updated tests for entity framework