Small fix so that the test case doesn't fail.