Fix build with updated drivers