Fix transition to input box