Fix test failing in Travis