Fixed up a test failure