Fix EntityTest using a mock Query.
Using appropriate variable name in EntityBuilder.