Small fix for Travis CI