Fixed keyboard focus being lost