fixed visualization bug