Improve integration spec

So it actually passes on Travis