diff --git a/tako/config.tako b/tako/config.tako
index 0728177..11f5a76 100644
--- a/tako/config.tako
+++ b/tako/config.tako
@@ -1,8 +1,4 @@
 import re
-# from tako.tools import COLORS_256
-
-COLORS_256 = {'COLOR_{}'.format(code): '\x1b[38;5;{}m'.format(code) for code in range(256)}
-COLORS_256['NO_COLOR'] = '\x1b[0m'
 
 $TAKO_SETTINGS.auto_pushd = True
 
@@ -46,8 +42,8 @@ else:
 def git_branch():
     ref=$(git symbolic-ref HEAD err> /dev/null)
     if not ref: return None
-    prefix = COLORS_256['COLOR_' + str(colors['gray'])] + ':'
-    branch = COLORS_256['COLOR_' + str(colors['orange'])] + re.sub('^refs/heads/', '', ref.strip())
+    prefix = colorize('gray', ':')
+    branch = colorize('orange', re.sub('^refs/heads/', '', ref[0].strip()))
     return prefix + branch
 
 $TAKO_SETTINGS.prompt_fields['git_branch'] = git_branch