Make sure getEntityByID has entity attribute