diff --git a/mm/huge_memory.c b/mm/huge_memory.c
index 794057aa7ee73c..5b38aa24566f19 100644
--- a/mm/huge_memory.c
+++ b/mm/huge_memory.c
@@ -2451,10 +2451,10 @@ static void collapse_huge_page(struct mm_struct *mm,
 		goto out;
 	}
 
-	anon_vma_lock_write(vma->anon_vma);
-
 	__collapse_huge_page_swapin(mm, vma, address, pmd);
 
+	anon_vma_lock_write(vma->anon_vma);
+
 	pte = pte_offset_map(pmd, address);
 	pte_ptl = pte_lockptr(mm, pmd);