Fix bug on key display