Fix CI script