Fixing input field issue