Fix for Travis CI