Fixed a bug with text visualisation