diff --git a/include/linux/swapops.h b/include/linux/swapops.h
index 0d429a102d417955192d610b37648966642f0b30..001a0f27515fb7339ae62ee6889a772d3b0efbef 100644
--- a/include/linux/swapops.h
+++ b/include/linux/swapops.h
@@ -281,7 +281,7 @@ static inline pmd_t swp_entry_to_pmd(swp_entry_t entry)
 
 static inline int is_pmd_migration_entry(pmd_t pmd)
 {
-	return !pmd_present(pmd) && is_migration_entry(pmd_to_swp_entry(pmd));
+	return is_swap_pmd(pmd) && is_migration_entry(pmd_to_swp_entry(pmd));
 }
 #else
 static inline void set_pmd_migration_entry(struct page_vma_mapped_walk *pvmw,