Attempt to fix tests in travis