diff --git a/kakoune/.config/kak/colors/four-colors-dark.kak b/kakoune/.config/kak/colors/four-colors-dark.kak
index bcb5c2a..28775d7 100644
--- a/kakoune/.config/kak/colors/four-colors-dark.kak
+++ b/kakoune/.config/kak/colors/four-colors-dark.kak
@@ -66,12 +66,14 @@ evaluate-commands %sh{
         set-face global Error $pink2
         set-face global DiagnosticError default,default,$pink2+c
         set-face global DiagnosticWarning default,default,$yellow2+c
+        set-face global DiagnosticInfo default,default,$green2+c
+        set-face global DiagnosticHint default,default,$blue2+c
         set-face global StatusLine $fg,$bgmenu
         set-face global StatusLineMode $green2
         set-face global StatusLineInfo $yellow2
         set-face global StatusLineValue $blue2
         set-face global StatusCursor $bg,$green1
-        set-face global Prompt $blue3
+        set-face global Prompt $green2
         set-face global BufferPadding $bg,$bg
 
         # Built-in highlighter faces
diff --git a/kakoune/.config/kak/colors/four-colors-light.kak b/kakoune/.config/kak/colors/four-colors-light.kak
index 74a54f0..f38db4b 100644
--- a/kakoune/.config/kak/colors/four-colors-light.kak
+++ b/kakoune/.config/kak/colors/four-colors-light.kak
@@ -66,12 +66,14 @@ evaluate-commands %sh{
         set-face global Error $pink2
         set-face global DiagnosticError default,default,$pink2+c
         set-face global DiagnosticWarning default,default,$yellow2+c
+        set-face global DiagnosticInfo default,default,$green2+c
+        set-face global DiagnosticHint default,default,$blue2+c
         set-face global StatusLine $fg,$bgmenu
         set-face global StatusLineMode $green2
         set-face global StatusLineInfo $yellow2
         set-face global StatusLineValue $blue2
         set-face global StatusCursor $bg,$green1
-        set-face global Prompt $blue3
+        set-face global Prompt $green2
         set-face global BufferPadding $bg,$bg
 
         # Built-in highlighter faces
diff --git a/kakoune/.config/kak/kakrc b/kakoune/.config/kak/kakrc
index c55a4d3..554eba7 100644
--- a/kakoune/.config/kak/kakrc
+++ b/kakoune/.config/kak/kakrc
@@ -15,14 +15,6 @@ nop %sh{
 	[ -e "$kak_config/autoload/stdlib" ] || ln -s "$kak_runtime/autoload" "$kak_config/autoload/stdlib"
 }
 
-colorscheme %sh{
-	if [ -e ~/.lightmode ]; then
-		echo four-colors-light
-	else
-		echo four-colors-dark
-	fi
-}
-
 declare-user-mode file-mode
 declare-user-mode options-mode
 
@@ -208,3 +200,13 @@ bundle 'peneira' 'https://github.com/gustavo-hms/peneira' %{
     map global -docstring 'Symbols' peneira-mode 's' ': peneira-symbols<ret>'
     map global -docstring 'Recent files' peneira-mode 'r' ': peneira-recent-files<ret>'
 }
+
+# LOCAL
+
+colorscheme %sh{
+	if [ -e ~/.lightmode ]; then
+		echo four-colors-light
+	else
+		echo four-colors-dark
+	fi
+}