Fix a demo bug