Fix display bugs