bug fix in visualization