Fixing demo code.