Fix CI issue