change the driver test code