diff --git a/arch/arm64/kvm/hyp/nvhe/pkvm.c b/arch/arm64/kvm/hyp/nvhe/pkvm.c
index 6443cfa..424f4b3 100644
--- a/arch/arm64/kvm/hyp/nvhe/pkvm.c
+++ b/arch/arm64/kvm/hyp/nvhe/pkvm.c
@@ -377,7 +377,7 @@ struct pkvm_hyp_vm *get_pkvm_hyp_vm(pkvm_handle_t handle)
 	if (hyp_vm->is_dying)
 		hyp_vm = NULL;
 	else
-		hyp_page_ref_inc(hyp_virt_to_page(hyp_vm));
+		hyp_refcount_inc(hyp_vm->refcount);
 
 unlock:
 	hyp_read_unlock(&vm_table_lock);
@@ -387,7 +387,7 @@ struct pkvm_hyp_vm *get_pkvm_hyp_vm(pkvm_handle_t handle)
 
 void put_pkvm_hyp_vm(struct pkvm_hyp_vm *hyp_vm)
 {
-	hyp_page_ref_dec(hyp_virt_to_page(hyp_vm));
+	hyp_refcount_dec(hyp_vm->refcount);
 }
 
 int __pkvm_reclaim_dying_guest_page(pkvm_handle_t handle, u64 pfn, u64 ipa)