Fix for travis.ci