drv_ctx->driver.shutdown = blkvsc_shutdown;
 
        /* The driver belongs to vmbus */
-       ret = vmbus_child_driver_register(drv_ctx);
+       ret = vmbus_child_driver_register(&drv_ctx->driver);
 
        return ret;
 }
 
        drv_ctx->driver.remove = mousevsc_remove;
 
        /* The driver belongs to vmbus */
-       vmbus_child_driver_register(drv_ctx);
+       vmbus_child_driver_register(&drv_ctx->driver);
 
        return 0;
 }
 
        drv_ctx->driver.remove = netvsc_remove;
 
        /* The driver belongs to vmbus */
-       ret = vmbus_child_driver_register(drv_ctx);
+       ret = vmbus_child_driver_register(&drv_ctx->driver);
 
        return ret;
 }
 
        drv_ctx->driver.remove = storvsc_remove;
 
        /* The driver belongs to vmbus */
-       ret = vmbus_child_driver_register(drv_ctx);
+       ret = vmbus_child_driver_register(&drv_ctx->driver);
 
        return ret;
 }
 
 
 
 /* Vmbus interface */
-int vmbus_child_driver_register(struct driver_context *driver_ctx);
+int vmbus_child_driver_register(struct device_driver *drv);
 void vmbus_child_driver_unregister(struct driver_context *driver_ctx);
 
 extern struct completion hv_channel_ready;
 
 
 /**
  * vmbus_child_driver_register() - Register a vmbus's child driver
- * @driver_ctx:        Pointer to driver structure you want to register
+ * @drv:        Pointer to driver structure you want to register
  *
- * @driver_ctx is of type &struct driver_context
  *
  * Registers the given driver with Linux through the 'driver_register()' call
  * And sets up the hyper-v vmbus handling for this driver.
  *
  * Mainly used by Hyper-V drivers.
  */
-int vmbus_child_driver_register(struct driver_context *driver_ctx)
+int vmbus_child_driver_register(struct device_driver *drv)
 {
        int ret;
 
        DPRINT_INFO(VMBUS_DRV, "child driver (%p) registering - name %s",
-                   driver_ctx, driver_ctx->driver.name);
+                   drv, drv->name);
 
        /* The child driver on this vmbus */
-       driver_ctx->driver.bus = &vmbus_drv.bus;
+       drv->bus = &vmbus_drv.bus;
 
-       ret = driver_register(&driver_ctx->driver);
+       ret = driver_register(drv);
 
        vmbus_request_offers();