Fix build with the new drivers