From e18f3fab751a4c8315561ddcfb37eb1342e94605 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kim=20Silkeb=C3=A6kken?= <kim.silkebaekken@gmail.com>
Date: Fri, 23 Nov 2012 19:03:58 +0100
Subject: [PATCH] Fix mode segment

---
 vim/segments/core.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/vim/segments/core.py b/vim/segments/core.py
index 60d98046..e5586d96 100644
--- a/vim/segments/core.py
+++ b/vim/segments/core.py
@@ -47,10 +47,13 @@ def mode(override=None):
 	'''
 	mode = vim_funcs['mode']()
 
+	if not override:
+		return (mode, vim_modes[mode])
+
 	try:
-		return override[mode]
-	except TypeError:
-		return vim_modes[mode]
+		return (mode, override[mode])
+	except IndexError:
+		return (mode, vim_modes[mode])
 
 
 def modified_indicator(text=u'+'):