Fix issue with input box