diff --git a/display.go b/display.go
index b512008..0769ba2 100644
--- a/display.go
+++ b/display.go
@@ -64,7 +64,11 @@ func DisplaySessionInfos(w io.Writer, sessions []SessionInfo, noColor bool, now
 			fmt.Fprintln(w)
 		}
 
-		stateStr := stateColorStr(s.State, noColor) + s.State + resetIfColor(noColor)
+		stateDisplay := s.State
+		if s.State == "Needs Input" && s.WaitType != "" {
+			stateDisplay = s.State + ": " + s.WaitType
+		}
+		stateStr := stateColorStr(s.State, noColor) + stateDisplay + resetIfColor(noColor)
 
 		branch := ""
 		if s.GitBranch != "" {