Fixed tests on Travis CI