PLF-5950 | Fix UT failed