BATCHEE-51 fix unit test