[IM] fix up unit tests for commit af4c487