fixed the demo error