1 files changed, 5 insertions, 5 deletions

diff --git a/mm/memcontrol.c b/mm/memcontrol.c
index e3cd40b2d5d9..f1a0ae6e11b8 100644
--- a/mm/memcontrol.c
+++ b/mm/memcontrol.c
@@ -6605,10 +6605,10 @@ static int mem_cgroup_count_precharge_pte_range(pmd_t *pmd,
         pte_t *pte;
         spinlock_t *ptl;
 
-        if (pmd_trans_huge_lock(pmd, vma) == 1) {
+        if (pmd_trans_huge_lock(pmd, vma, &ptl) == 1) {
                 if (get_mctgt_type_thp(vma, addr, *pmd, NULL) == MC_TARGET_PAGE)
                         mc.precharge += HPAGE_PMD_NR;
-                spin_unlock(&vma->vm_mm->page_table_lock);
+                spin_unlock(ptl);
                 return 0;
         }
 
@@ -6797,9 +6797,9 @@ static int mem_cgroup_move_charge_pte_range(pmd_t *pmd,
          *    to be unlocked in __split_huge_page_splitting(), where the main
          *    part of thp split is not executed yet.
          */
-        if (pmd_trans_huge_lock(pmd, vma) == 1) {
+        if (pmd_trans_huge_lock(pmd, vma, &ptl) == 1) {
                 if (mc.precharge < HPAGE_PMD_NR) {
-                        spin_unlock(&vma->vm_mm->page_table_lock);
+                        spin_unlock(ptl);
                         return 0;
                 }
                 target_type = get_mctgt_type_thp(vma, addr, *pmd, &target);
@@ -6816,7 +6816,7 @@ static int mem_cgroup_move_charge_pte_range(pmd_t *pmd,
                         }
                         put_page(page);
                 }
-                spin_unlock(&vma->vm_mm->page_table_lock);
+                spin_unlock(ptl);
                 return 0;
         }