1 files changed, 2 insertions, 1 deletions

diff --git a/include/asm-s390/processor.h b/include/asm-s390/processor.h
index c86b982aef5a..4f744609cd11 100644
--- a/include/asm-s390/processor.h
+++ b/include/asm-s390/processor.h
@@ -70,8 +70,9 @@ extern int get_cpu_capability(unsigned int *);
 
 #else /* __s390x__ */
 
-# define TASK_SIZE              (test_thread_flag(TIF_31BIT) ? \
+# define TASK_SIZE_OF(tsk)      (test_tsk_thread_flag(tsk, TIF_31BIT) ? \
                                         (0x80000000UL) : (0x40000000000UL))
+# define TASK_SIZE              TASK_SIZE_OF(current)
 # define TASK_UNMAPPED_BASE     (TASK_SIZE / 2)
 # define DEFAULT_TASK_SIZE      (0x40000000000UL)