Fix hide keyboard after button press bug.