Fixed bug in display