CI test fixes