diff --git a/arch/arm64/include/asm/kvm_pkvm_module.h b/arch/arm64/include/asm/kvm_pkvm_module.h
index 7d0b680d38e1..8b45704529e7 100644
--- a/arch/arm64/include/asm/kvm_pkvm_module.h
+++ b/arch/arm64/include/asm/kvm_pkvm_module.h
@@ -3,9 +3,13 @@
 #ifndef __ARM64_KVM_PKVM_MODULE_H__
 #define __ARM64_KVM_PKVM_MODULE_H__
 
+#include <asm/kvm_pgtable.h>
 #include <linux/export.h>
 
 struct pkvm_module_ops {
+	int (*create_private_mapping)(phys_addr_t phys, size_t size,
+				      enum kvm_pgtable_prot prot,
+				      unsigned long *haddr);
 };
 
 struct pkvm_module_section {
diff --git a/arch/arm64/kvm/hyp/nvhe/modules.c b/arch/arm64/kvm/hyp/nvhe/modules.c
index 9e695d9a4243..43c670d4f7cf 100644
--- a/arch/arm64/kvm/hyp/nvhe/modules.c
+++ b/arch/arm64/kvm/hyp/nvhe/modules.c
@@ -6,8 +6,10 @@
 #include <asm/kvm_pkvm_module.h>
 
 #include <nvhe/modules.h>
+#include <nvhe/mm.h>
 
 const struct pkvm_module_ops module_ops = {
+	.create_private_mapping = __pkvm_create_private_mapping,
 };
 
 int __pkvm_init_module(void *module_init)