Small fix for demo
Merge pull request #75 from DimaSoroka/master

Small fix for demo