Improve ime handling