fix entity not returning