Fixed a bug in the visualiser.