fixing vizualization bug