Fix test failure