PIM-1370 : Update UT