Fix display bug (?)