Fixed bugs for demo.