Small fix on demo.