{
        struct nvmem_device *nvmem = to_nvmem_device(dev);
 
-       ida_simple_remove(&nvmem_ida, nvmem->id);
+       ida_free(&nvmem_ida, nvmem->id);
        gpiod_put(nvmem->wp_gpio);
        kfree(nvmem);
 }
        if (!nvmem)
                return ERR_PTR(-ENOMEM);
 
-       rval  = ida_simple_get(&nvmem_ida, 0, 0, GFP_KERNEL);
+       rval  = ida_alloc(&nvmem_ida, GFP_KERNEL);
        if (rval < 0) {
                kfree(nvmem);
                return ERR_PTR(rval);
                nvmem->wp_gpio = gpiod_get_optional(config->dev, "wp",
                                                    GPIOD_OUT_HIGH);
        if (IS_ERR(nvmem->wp_gpio)) {
-               ida_simple_remove(&nvmem_ida, nvmem->id);
+               ida_free(&nvmem_ida, nvmem->id);
                rval = PTR_ERR(nvmem->wp_gpio);
                kfree(nvmem);
                return ERR_PTR(rval);