1 files changed, 3 insertions, 2 deletions

diff --git a/arch/s390/mm/pgtable.c b/arch/s390/mm/pgtable.c
index 37a23c22370..458893f5f6b 100644
--- a/arch/s390/mm/pgtable.c
+++ b/arch/s390/mm/pgtable.c
@@ -291,8 +291,9 @@ void page_table_free_rcu(struct mmu_gather *tlb, unsigned long *table)
 
 void __tlb_remove_table(void *_table)
 {
-        void *table = (void *)((unsigned long) _table & PAGE_MASK);
-        unsigned type = (unsigned long) _table & ~PAGE_MASK;
+        const unsigned long mask = (FRAG_MASK << 4) | FRAG_MASK;
+        void *table = (void *)((unsigned long) _table & ~mask);
+        unsigned type = (unsigned long) _table & mask;
 
         if (type)
                 __page_table_free_rcu(table, type);