diff --git a/layer2img.py b/layer2img.py
index f58b28c..b9fcc3f 100644
--- a/layer2img.py
+++ b/layer2img.py
@@ -18,9 +18,7 @@ TEXTWIDTH_CM = 12.12364
 
 
 def color(string):
-    if not string:
-        string = 'black'
-    return COLORS[string]
+    return COLORS[string if string else 'black']
 
 
 def inch(cm):