Fix the driver