1 files changed, 3 insertions, 3 deletions

diff --git a/mm/huge_memory.c b/mm/huge_memory.c
index d7ee1691fd21..32754eece63e 100644
--- a/mm/huge_memory.c
+++ b/mm/huge_memory.c
@@ -1460,9 +1460,10 @@ int change_huge_pmd(struct vm_area_struct *vma, pmd_t *pmd,
         if (__pmd_trans_huge_lock(pmd, vma) == 1) {
                 pmd_t entry;
                 entry = pmdp_get_and_clear(mm, addr, pmd);
-                if (!prot_numa)
+                if (!prot_numa) {
                         entry = pmd_modify(entry, newprot);
-                else {
+                        BUG_ON(pmd_write(entry));
+                } else {
                         struct page *page = pmd_page(*pmd);
 
                         /* only check non-shared pages */
@@ -1471,7 +1472,6 @@ int change_huge_pmd(struct vm_area_struct *vma, pmd_t *pmd,
                                 entry = pmd_mknuma(entry);
                         }
                 }
-                BUG_ON(pmd_write(entry));
                 set_pmd_at(mm, addr, pmd, entry);
                 spin_unlock(&vma->vm_mm->page_table_lock);
                 ret = 1;