link++;
        }
 
-       kfree(ctx->links);
-       ctx->links = NULL;
-
        return 0;
 }
 
 
        dev_dbg(&adev->dev, "Creating %d SDW Link devices\n", count);
 
-       ctx = kzalloc(sizeof(*ctx), GFP_KERNEL);
+       ctx = devm_kzalloc(&adev->dev, sizeof(*ctx), GFP_KERNEL);
        if (!ctx)
                return NULL;
 
        ctx->count = count;
-       ctx->links = kcalloc(ctx->count, sizeof(*ctx->links), GFP_KERNEL);
+       ctx->links = devm_kcalloc(&adev->dev, ctx->count,
+                                 sizeof(*ctx->links), GFP_KERNEL);
        if (!ctx->links)
-               goto link_err;
+               return NULL;
 
        link = ctx->links;
 
        return ctx;
 
 pdev_err:
+       ctx->count = i;
        sdw_intel_cleanup_pdev(ctx);
-link_err:
-       kfree(ctx);
        return NULL;
 }
 
 void sdw_intel_exit(struct sdw_intel_ctx *ctx)
 {
        sdw_intel_cleanup_pdev(ctx);
-       kfree(ctx);
 }
 EXPORT_SYMBOL(sdw_intel_exit);