PIM-1544 : Fix UT