Fix failure on CI