Fix Display bug