Fixed a bug in the display