SOC-3808 | Fix UT failed