1 files changed, 10 insertions, 0 deletions
diff --git a/test/test_vm_page_fill.c b/test/test_vm_page_fill.c
index 2b7f1130..0b4278bf 100644
--- a/test/test_vm_page_fill.c
+++ b/test/test_vm_page_fill.c
@@ -45,10 +45,15 @@ static unsigned char test_pattern = 1;
 static void
 test_write_pages(void)
 {
+    struct vm_map *kernel_map;
+    struct pmap *kernel_pmap;
     struct vm_page *page;
     int error, flags;
     uintptr_t va;
 
+    kernel_map = vm_map_get_kernel_map();
+    kernel_pmap = pmap_get_kernel_pmap();
+
     for (;;) {
         page = vm_page_alloc(0, VM_PAGE_SEL_HIGHMEM, VM_PAGE_KERNEL);
 
@@ -80,10 +85,15 @@ test_write_pages(void)
 static void
 test_reset_pages(void)
 {
+    struct vm_map *kernel_map;
+    struct pmap *kernel_pmap;
     struct vm_page *page;
     int error, flags;
     uintptr_t va;
 
+    kernel_map = vm_map_get_kernel_map();
+    kernel_pmap = pmap_get_kernel_pmap();
+
     while (!list_empty(&test_pages)) {
         page = list_first_entry(&test_pages, struct vm_page, node);
         list_remove(&page->node);