Fixed minor issue with the visualization window