diff --git a/mm/swap.c b/mm/swap.c
index 7d7f80c..7782588 100644
--- a/mm/swap.c
+++ b/mm/swap.c
@@ -258,8 +258,7 @@
 		if (WARN_ON(kiov[seg].iov_len != PAGE_SIZE))
 			return seg;
 
-		/* virt_to_page sanity checks the PFN */
-		pages[seg] = virt_to_page(kiov[seg].iov_base);
+		pages[seg] = kmap_to_page(kiov[seg].iov_base);
 		page_cache_get(pages[seg]);
 	}