fix little bug in demo