SOC-3430 | Fix UT failed