change for new method of getting display entity