1 files changed, 2 insertions, 2 deletions

diff --git a/runlist.c b/runlist.c
index 2eee01c..22f47ff 100644
--- a/runlist.c
+++ b/runlist.c
@@ -64,9 +64,9 @@ int get_runlist_iter(struct nvdebug_state *g, int rl_id, struct runlist_iter *rl
                         goto attempt_pramin_access;
 
                 if (pd_config.is_ver2)
-                        runlist_bar_vaddr = search_page_directory(g, pd_config, runlist_iova);
+                        runlist_bar_vaddr = search_page_directory(g, pd_config, runlist_iova, TARGET_VID_MEM);
                 else
-                        runlist_bar_vaddr = search_v1_page_directory(g, pd_config, runlist_iova);
+                        runlist_bar_vaddr = search_v1_page_directory(g, pd_config, runlist_iova, TARGET_VID_MEM);
                 if (!runlist_bar_vaddr) {
                         printk(KERN_WARNING "[nvdebug] Unable to find runlist %d mapping in BAR2/3 page tables for %x.\n", rl_id, g->chip_id);
                         err = -EOPNOTSUPP;