Fix CI problem