Fix ci tests