Fixed demo to support v1 of framework