Fixed two IM unit tests