Fixed failing test as we had new stub data