fixing another visualization bug similar to the one before