Fix bug when using ime