Fixed visualization #2