Fix text box issue