Fix visualization.