correct visualization bug