Display bug fix