React to EF change in tests