Fixed the demo