Fix input box bug