[CI] Improve setup script