Update drivers to the API changes