static void nodemgr_remove_uds(struct node_entry *ne)
 {
        struct class_device *cdev;
-       struct unit_directory *ud, **unreg;
-       size_t i, count;
+       struct unit_directory *tmp, *ud;
 
-       /*
-        * This is awkward:
-        * Iteration over nodemgr_ud_class.children has to be protected by
+       /* Iteration over nodemgr_ud_class.children has to be protected by
         * nodemgr_ud_class.sem, but class_device_unregister() will eventually
-        * take nodemgr_ud_class.sem too. Therefore store all uds to be
-        * unregistered in a temporary array, release the semaphore, and then
-        * unregister the uds.
-        *
-        * Since nodemgr_remove_uds can also run in other contexts than the
-        * knodemgrds (which are currently globally serialized), protect the
+        * take nodemgr_ud_class.sem too. Therefore pick out one ud at a time,
+        * release the semaphore, and then unregister the ud. Since this code
+        * may be called from other contexts besides the knodemgrds, protect the
         * gap after release of the semaphore by nodemgr_serialize_remove_uds.
         */
-
        mutex_lock(&nodemgr_serialize_remove_uds);
-
-       down(&nodemgr_ud_class.sem);
-       count = 0;
-       list_for_each_entry(cdev, &nodemgr_ud_class.children, node) {
-               ud = container_of(cdev, struct unit_directory, class_dev);
-               if (ud->ne == ne)
-                       count++;
-       }
-       if (!count) {
-               up(&nodemgr_ud_class.sem);
-               mutex_unlock(&nodemgr_serialize_remove_uds);
-               return;
-       }
-       unreg = kcalloc(count, sizeof(*unreg), GFP_KERNEL);
-       if (!unreg) {
-               HPSB_ERR("NodeMgr: out of memory in nodemgr_remove_uds");
-               up(&nodemgr_ud_class.sem);
-               mutex_unlock(&nodemgr_serialize_remove_uds);
-               return;
-       }
-       i = 0;
-       list_for_each_entry(cdev, &nodemgr_ud_class.children, node) {
-               ud = container_of(cdev, struct unit_directory, class_dev);
-               if (ud->ne == ne) {
-                       BUG_ON(i >= count);
-                       unreg[i++] = ud;
+       for (;;) {
+               ud = NULL;
+               down(&nodemgr_ud_class.sem);
+               list_for_each_entry(cdev, &nodemgr_ud_class.children, node) {
+                       tmp = container_of(cdev, struct unit_directory,
+                                          class_dev);
+                       if (tmp->ne == ne) {
+                               ud = tmp;
+                               break;
+                       }
                }
+               up(&nodemgr_ud_class.sem);
+               if (ud == NULL)
+                       break;
+               class_device_unregister(&ud->class_dev);
+               device_unregister(&ud->device);
        }
-       up(&nodemgr_ud_class.sem);
-
-       for (i = 0; i < count; i++) {
-               class_device_unregister(&unreg[i]->class_dev);
-               device_unregister(&unreg[i]->device);
-       }
-       kfree(unreg);
-
        mutex_unlock(&nodemgr_serialize_remove_uds);
 }