small visualisation improvement