Updated demo to reflect latest api-change