updated the test code.