Fix the demo app