Fix awkward keyboard focus