Fix minor test failure on CI