SOC-3972 | Fix UT failed