modification of the test code