1 files changed, 2 insertions, 2 deletions

diff --git a/arch/s390/include/asm/tlbflush.h b/arch/s390/include/asm/tlbflush.h
index 1d8648cf2fea..9fde315f3a7c 100644
--- a/arch/s390/include/asm/tlbflush.h
+++ b/arch/s390/include/asm/tlbflush.h
@@ -27,12 +27,12 @@ static inline void __tlb_flush_global(void)
         register unsigned long reg4 asm("4");
         long dummy;
 
-#ifndef __s390x__
+#ifndef CONFIG_64BIT
         if (!MACHINE_HAS_CSP) {
                 smp_ptlb_all();
                 return;
         }
-#endif /* __s390x__ */
+#endif /* CONFIG_64BIT */
 
         dummy = 0;
         reg2 = reg3 = 0;