SOC-4377 | Fix UT failed