Fix CI setup