NEXUS-6164: Fixed UT incomplete mock