Fixed bug in demo file