fixes test for entities