Merge "Fix IME callback not being re-registered on screen off / on." into tm-dev