fix visual bugs