1 files changed, 3 insertions, 3 deletions

diff --git a/arch/s390/include/asm/pgalloc.h b/arch/s390/include/asm/pgalloc.h
index f0f9bcf94c03..5ee733720a57 100644
--- a/arch/s390/include/asm/pgalloc.h
+++ b/arch/s390/include/asm/pgalloc.h
@@ -36,11 +36,11 @@ static inline void crst_table_init(unsigned long *crst, unsigned long entry)
 
 static inline unsigned long pgd_entry_type(struct mm_struct *mm)
 {
-        if (mm->context.asce_limit <= _REGION3_SIZE)
+        if (mm_pmd_folded(mm))
                 return _SEGMENT_ENTRY_EMPTY;
-        if (mm->context.asce_limit <= _REGION2_SIZE)
+        if (mm_pud_folded(mm))
                 return _REGION3_ENTRY_EMPTY;
-        if (mm->context.asce_limit <= _REGION1_SIZE)
+        if (mm_p4d_folded(mm))
                 return _REGION2_ENTRY_EMPTY;
         return _REGION1_ENTRY_EMPTY;
 }