diff --git a/src/backend.rs b/src/backend.rs
index 4748d6c..4638a17 100644
--- a/src/backend.rs
+++ b/src/backend.rs
@@ -30,6 +30,7 @@ use tower_lsp::lsp_types::{
 use tower_lsp::{jsonrpc::Result, lsp_types::WorkspaceServerCapabilities};
 use tower_lsp::{Client, LanguageServer};
 use tremor_script::arena::Arena;
+use tremor_script::highlighter::ErrorLevel;
 
 // stores the latest state of the document as it changes (on edits)
 // TODO can add more fields here based on ast parsing
@@ -86,6 +87,10 @@ impl Backend {
                     message = format!("{}, Note: {}", message, hint);
                 }
 
+                if let ErrorLevel::Warning(class) = e.level() {
+                    message = format!("{}: {}", class, message);
+                }
+
                 diagnostics.push(Diagnostic {
                     range,
                     message,