HIVE-5871: fix UT