Quick fix

quick fix to the test file.