HIVE-8017: fix UT