fix for visualisation