Update the tests to work locally and using Travis CI