fixed bugs for demo.