fix the input problem