Fix a small bug in visualization.