PIM-1530: Clean UT