Fix testing on Travis CI