Updates demo to work in latest beta.