Update test function