Fixed bug in demo application