Fixed the tests on travis