Visualizer: Minor bug fix