Fix display issues caused by last update