Update tests for Travis CI