Improve robustness of correct function to handle chinese