Fixed a bug in the demo.