Update driver imports