1 files changed, 1 insertions, 4 deletions

diff --git a/arch/s390/mm/hugetlbpage.c b/arch/s390/mm/hugetlbpage.c
index d261c62e40a6..0727a55d87d9 100644
--- a/arch/s390/mm/hugetlbpage.c
+++ b/arch/s390/mm/hugetlbpage.c
@@ -123,10 +123,7 @@ pte_t huge_ptep_get_and_clear(struct mm_struct *mm,
         pmd_t *pmdp = (pmd_t *) ptep;
         pte_t pte = huge_ptep_get(ptep);
 
-        if (MACHINE_HAS_IDTE)
-                __pmd_idte(addr, pmdp);
-        else
-                __pmd_csp(pmdp);
+        pmdp_flush_direct(mm, addr, pmdp);
         pmd_val(*pmdp) = _SEGMENT_ENTRY_EMPTY;
         return pte;
 }