Fix input focus problem