Fix bug in display