Visualization fix