Fix the keyboard mode after upgrades as well.