Fixed bug in demo.