Small fix to the test code