diff --git a/arch/arm64/kvm/hyp/nvhe/mem_protect.c b/arch/arm64/kvm/hyp/nvhe/mem_protect.c
index bd9ae1bc37f1..01c43037447a 100644
--- a/arch/arm64/kvm/hyp/nvhe/mem_protect.c
+++ b/arch/arm64/kvm/hyp/nvhe/mem_protect.c
@@ -1076,7 +1076,9 @@ static int host_ack_donation(u64 addr, const struct pkvm_mem_transition *tx)
 
 static int host_ack_unshare(u64 addr, const struct pkvm_mem_transition *tx)
 {
-	return __host_ack_transition(addr, tx, PKVM_PAGE_SHARED_BORROWED);
+	u64 size = tx->nr_pages * PAGE_SIZE;
+
+	return __host_check_page_state_range(addr, size, PKVM_PAGE_SHARED_BORROWED);
 }
 
 static int host_complete_share(u64 addr, const struct pkvm_mem_transition *tx,