Fix test failing