fix the demo