SOC-4401 | Fix UT failed