changes to the test code