diff --git a/drivers/infiniband/hw/hns/hns_roce_hem.c b/drivers/infiniband/hw/hns/hns_roce_hem.c
index a8dccba39927cffa262fba6d2358a4c657e58e8c..5b5272f8298e5e5b6fb807f99b7fc82e41bf61d6 100644
--- a/drivers/infiniband/hw/hns/hns_roce_hem.c
+++ b/drivers/infiniband/hw/hns/hns_roce_hem.c
@@ -277,7 +277,7 @@ static struct hns_roce_hem *hns_roce_alloc_hem(struct hns_roce_dev *hr_dev,
 	return hem;
 
 fail:
-	kfree(hem);
+	hns_roce_free_hem(hr_dev, hem);
 	return NULL;
 }