fix small viz bug