updates for the test code