Fix test failure?