Fix ITM UT failures