Trying to fix tests on travis