Fixing Travis test