fix driver code