diff --git a/mm/swap_state.c b/mm/swap_state.c
index c1a648d9092bd311cdc82aa1fe5648d9f4f7c03b..f270c30d4681368333c88aeea546d92ad1607b6f 100644
--- a/mm/swap_state.c
+++ b/mm/swap_state.c
@@ -87,11 +87,9 @@ void *get_shadow_from_swap_cache(swp_entry_t entry)
 	pgoff_t idx = swp_offset(entry);
 	struct page *page;
 
-	page = find_get_entry(address_space, idx);
+	page = xa_load(&address_space->i_pages, idx);
 	if (xa_is_value(page))
 		return page;
-	if (page)
-		put_page(page);
 	return NULL;
 }