Update demo for API change