Small fix due to display changes.