Update driver code imports in the driver itself