int i = 0;
 
        lhea_dn = adapter->ebus_dev->ofdev.node;
-       do {
-               eth_dn = of_get_next_child(lhea_dn, eth_dn);
-               if (!eth_dn)
-                       break;
-
+       while ((eth_dn = of_get_next_child(lhea_dn, eth_dn))) {
+               
                dn_log_port_id = (u32*)get_property(eth_dn, "ibm,hea-port-no",
                                                    NULL);
                if (!dn_log_port_id) {
                                  adapter->port[i]->netdev->name, 
                                  *dn_log_port_id);
                i++;
-       } while (eth_dn);
-
-       of_node_put(lhea_dn);
+       };
 
        /* Check for succesfully set up ports */
        for (i = 0; i < EHEA_MAX_PORTS; i++)
        u32 *dn_log_port_id;
 
        lhea_dn = adapter->ebus_dev->ofdev.node;
-       do {
-               eth_dn = of_get_next_child(lhea_dn, eth_dn);
-               if (!eth_dn)
-                       break;
-
+       while ((eth_dn = of_get_next_child(lhea_dn, eth_dn))) {
+               
                dn_log_port_id = (u32*)get_property(eth_dn, "ibm,hea-port-no",
                                                    NULL);
-
                if (dn_log_port_id)
                        if (*dn_log_port_id == logical_port_id)
                                return eth_dn;
-
-       } while (eth_dn);
-
-       of_node_put(lhea_dn);
+       };
 
        return NULL;
 }
                
        port = ehea_setup_single_port(adapter, logical_port_id, eth_dn);
 
+       of_node_put(eth_dn);
+
        if (port) {
                for (i=0; i < EHEA_MAX_PORTS; i++)
                        if (!adapter->port[i]) {
        u64 *adapter_handle;
        int ret;
 
+       if (!dev || !dev->ofdev.node) {
+               ehea_error("Invalid ibmebus device probed");
+               return -EINVAL;
+       }
+
        adapter = kzalloc(sizeof(*adapter), GFP_KERNEL);
        if (!adapter) {
                ret = -ENOMEM;
        adapter->neq = ehea_create_eq(adapter,
                                      EHEA_NEQ, EHEA_MAX_ENTRIES_EQ, 1);
        if (!adapter->neq) {
+               ret = -EIO;
                dev_err(&dev->ofdev.dev, "NEQ creation failed");
                goto out_free_res;
        }
        }
 
        adapter->ehea_wq = create_workqueue("ehea_wq");
-       if (!adapter->ehea_wq)
+       if (!adapter->ehea_wq) {
+               ret = -EIO;
                goto out_free_irq;
+       }
 
-       if (ehea_create_device_sysfs(dev))
+       ret = ehea_create_device_sysfs(dev);
+       if (ret)
                goto out_kill_wq;
 
        ret = ehea_setup_ports(adapter);