Update tests to work on TravisCI