diff --git a/vim/.vim/lua/plugins/diagnostics.lua b/vim/.vim/lua/plugins/diagnostics.lua
index 381ec29..b7fda08 100644
--- a/vim/.vim/lua/plugins/diagnostics.lua
+++ b/vim/.vim/lua/plugins/diagnostics.lua
@@ -34,6 +34,9 @@ return {
 				local new_value = not vim.diagnostic.config().virtual_lines.only_current_line
 				vim.diagnostic.config({
 					virtual_text = not new_value,
+					virtual_improved = {
+						current_line = new_value and "hide" or "show",
+					},
 					virtual_lines = { only_current_line = new_value },
 				})
 				return new_value