}
 
        /* Allocate memory for MTD device structure and private data */
-       new_mtd = kzalloc(sizeof(struct mtd_info) + sizeof(struct nand_chip), GFP_KERNEL);
-       if (!new_mtd) {
+       this = kzalloc(sizeof(struct nand_chip), GFP_KERNEL);
+       if (!this) {
                err = -ENOMEM;
                goto out;
        }
 
-       /* Get pointer to private data */
-       this = (struct nand_chip *)(&new_mtd[1]);
+       new_mtd = nand_to_mtd(this);
 
        /* Link the private data with the MTD structure */
        new_mtd->priv = this;
 out_ior:
        iounmap(this->IO_ADDR_R);
 out_mtd:
-       kfree(new_mtd);
+       kfree(this);
 out:
        return err;
 }
                if (!mtd)
                        continue;
 
-               this = mtd_to_nand(cs553x_mtd[i]);
+               this = mtd_to_nand(mtd);
                mmio_base = this->IO_ADDR_R;
 
                /* Release resources, unregister device */
-               nand_release(cs553x_mtd[i]);
-               kfree(cs553x_mtd[i]->name);
+               nand_release(mtd);
+               kfree(mtd->name);
                cs553x_mtd[i] = NULL;
 
                /* unmap physical address */
                iounmap(mmio_base);
 
                /* Free the MTD device structure */
-               kfree(mtd);
+               kfree(this);
        }
 }