Use the right entity object