visualisation bug fix