fixed a small issue in demo