Tweak CI script