Fix terrible text focus bug