Quick update for test failure