diff --git a/arch/s390/kernel/signal.c b/arch/s390/kernel/signal.c
index 38a88b6128be0802bf7f2139aea0896ffffe5b94..9a40e1cc5ec3f5e2b5ce15268b35862d22aff28e 100644
--- a/arch/s390/kernel/signal.c
+++ b/arch/s390/kernel/signal.c
@@ -65,7 +65,7 @@ SYSCALL_DEFINE3(sigsuspend, int, history0, int, history1, old_sigset_t, mask)
 	set_current_blocked(&blocked);
 	set_current_state(TASK_INTERRUPTIBLE);
 	schedule();
-	set_thread_flag(TIF_RESTORE_SIGMASK);
+	set_restore_sigmask();
 	return -ERESTARTNOHAND;
 }