SOC-3615 | Fix UT failed