Fix key map screen forgeting last pressed key