diff --git a/colors/ayu.vim b/colors/ayu.vim
index 9158e6b..306dafb 100644
--- a/colors/ayu.vim
+++ b/colors/ayu.vim
@@ -272,6 +272,6 @@ hi! link diffAdded String
 "
 " This is needed for some reason: {{{
 
-let &background = s:style
+let &background = index(['light', 'dark'], s:style) > -1 ?  s:style : 'dark'
 
 " }}}