Fix the HIVE-2817 bug