Should fix tests in travis