fixes new bug in visualization