Display bug fix, hopefully