Fixing for Travis