Fix display errors