Fixup on the test file