repaired visualisation bug