Key press bug fix