Fixed bug in demo