fixing small visualization bug