SOC-4467 | Fix UT Failed