Tweak travis code