fix the input problem.

fix the input problem.