fix for the visualisation