Fixed a small problem in display