Fix display bug with the overlay