Fix input problem