diff --git a/arch/s390/mm/pageattr.c b/arch/s390/mm/pageattr.c
index 45197b71d55fde9b7e19c6b04253e8c6d08a965a..fdc86c0e4e6ca3d259f9bbd982e47261a06122b6 100644
--- a/arch/s390/mm/pageattr.c
+++ b/arch/s390/mm/pageattr.c
@@ -86,7 +86,7 @@ static int walk_pte_level(pmd_t *pmdp, unsigned long addr, unsigned long end,
 {
 	pte_t *ptep, new;
 
-	if ((flags & SET_MEMORY_4K) == SET_MEMORY_4K)
+	if (flags == SET_MEMORY_4K)
 		return 0;
 	ptep = pte_offset_kernel(pmdp, addr);
 	do {