fixed visualization bug in demo