correct a bug in visualisation;