fix flow visualization