Fix input behavior