Fix the demo