fixed another bug in the demo.