update test driver code