Fix test breaks on CI.