Fixed test error, Test entity ID was the source of it