Fix widget call in Entity