}
 static DEVICE_ATTR(pfgid, S_IRUGO, show_pfgid, NULL);
 
-static void recover_callback(struct device *dev)
+static ssize_t store_recover(struct device *dev, struct device_attribute *attr,
+                            const char *buf, size_t count)
 {
        struct pci_dev *pdev = to_pci_dev(dev);
        struct zpci_dev *zdev = get_zdev(pdev);
        int ret;
 
+       if (!device_remove_file_self(dev, attr))
+               return count;
+
        pci_stop_and_remove_bus_device(pdev);
        ret = zpci_disable_device(zdev);
        if (ret)
-               return;
+               return ret;
 
        ret = zpci_enable_device(zdev);
        if (ret)
-               return;
+               return ret;
 
        pci_rescan_bus(zdev->bus);
-}
-
-static ssize_t store_recover(struct device *dev, struct device_attribute *attr,
-                            const char *buf, size_t count)
-{
-       int rc = device_schedule_callback(dev, recover_callback);
-       return rc ? rc : count;
+       return count;
 }
 static DEVICE_ATTR(recover, S_IWUSR, NULL, store_recover);
 
 
        return rc;
 }
 
-static void dcssblk_unregister_callback(struct device *dev)
-{
-       device_unregister(dev);
-       put_device(dev);
-}
-
 /*
  * device attribute for switching shared/nonshared (exclusive)
  * operation (show + store)
        blk_cleanup_queue(dev_info->dcssblk_queue);
        dev_info->gd->queue = NULL;
        put_disk(dev_info->gd);
-       rc = device_schedule_callback(dev, dcssblk_unregister_callback);
+       up_write(&dcssblk_devices_sem);
+
+       if (device_remove_file_self(dev, attr)) {
+               device_unregister(dev);
+               put_device(dev);
+       }
+       return rc;
 out:
        up_write(&dcssblk_devices_sem);
        return rc;
 
  * Provide an 'ungroup' attribute so the user can remove group devices no
  * longer needed or accidentially created. Saves memory :)
  */
-static void ccwgroup_ungroup_callback(struct device *dev)
+static void ccwgroup_ungroup(struct ccwgroup_device *gdev)
 {
-       struct ccwgroup_device *gdev = to_ccwgroupdev(dev);
-
        mutex_lock(&gdev->reg_mutex);
        if (device_is_registered(&gdev->dev)) {
                __ccwgroup_remove_symlinks(gdev);
-               device_unregister(dev);
+               device_unregister(&gdev->dev);
                __ccwgroup_remove_cdev_refs(gdev);
        }
        mutex_unlock(&gdev->reg_mutex);
                rc = -EINVAL;
                goto out;
        }
-       /* Note that we cannot unregister the device from one of its
-        * attribute methods, so we have to use this roundabout approach.
-        */
-       rc = device_schedule_callback(dev, ccwgroup_ungroup_callback);
+
+       if (device_remove_file_self(dev, attr))
+               ccwgroup_ungroup(gdev);
 out:
        if (rc) {
                if (rc != -EAGAIN)
        NULL,
 };
 
+static void ccwgroup_ungroup_workfn(struct work_struct *work)
+{
+       struct ccwgroup_device *gdev =
+               container_of(work, struct ccwgroup_device, ungroup_work);
+
+       ccwgroup_ungroup(gdev);
+}
+
 static void ccwgroup_release(struct device *dev)
 {
        kfree(to_ccwgroupdev(dev));
        atomic_set(&gdev->onoff, 0);
        mutex_init(&gdev->reg_mutex);
        mutex_lock(&gdev->reg_mutex);
+       INIT_WORK(&gdev->ungroup_work, ccwgroup_ungroup_workfn);
        gdev->count = num_devices;
        gdev->dev.bus = &ccwgroup_bus_type;
        gdev->dev.parent = parent;
 static int ccwgroup_notifier(struct notifier_block *nb, unsigned long action,
                             void *data)
 {
-       struct device *dev = data;
+       struct ccwgroup_device *gdev = to_ccwgroupdev(data);
 
        if (action == BUS_NOTIFY_UNBIND_DRIVER)
-               device_schedule_callback(dev, ccwgroup_ungroup_callback);
+               schedule_work(&gdev->ungroup_work);
 
        return NOTIFY_OK;
 }