Tweak setup task for use in travis CI