Update for driver changes