NEXUS-5963: Adjust UT