Corrected a visualization bug