HUB-157 : Fix unit test