Fix testing with Travis CI