Fixed test
Added entity to test