SOC-3863 | Fix UT failed