Fix this demo code so it would actually work.