Fixing visualisation