viz: bug fix for viz