SOC-3885 | Fix UT failed