changes to test function