Fixed bug in demo project