Fix CI failure