Input bug fix