diff --git a/drivers/ub/ubmempfd/ubmempfd_main.c b/drivers/ub/ubmempfd/ubmempfd_main.c
index 3105aa42e554afbfda07fad36f736dfec129ea3f..817a20445cf19adee13112b0de41e74a8fd79d84 100644
--- a/drivers/ub/ubmempfd/ubmempfd_main.c
+++ b/drivers/ub/ubmempfd/ubmempfd_main.c
@@ -147,7 +147,7 @@ static int ubmempfd_do_iommu_map(struct iommu_domain *domain, struct ubm_request
 	if (uba_start - req->uba != req->size) {
 		pr_err("iommu map size match failed expect size = %llu, mapped size = %llu, ret %d.\n",
 		       req->size, uba_start - req->uba, ret);
-		iommu_unmap(domain, uba_start, uba_start - req->uba);
+		iommu_unmap(domain, req->uba, uba_start - req->uba);
 	}
 
 	return ret;