Fix unrelated test failure