diff --git a/lib/main.js b/lib/main.js
index 2db6589..1d54b82 100644
--- a/lib/main.js
+++ b/lib/main.js
@@ -51,9 +51,10 @@ export default {
         let match = regex.exec(output);
         while (match !== null) {
           const msgLine = Number.parseInt(match[1] - 1, 10);
+          const type = match[2] === 'warning' ? 'Warning' : 'Error';
           toReturn.push({
             range: helpers.rangeFromLineNumber(textEditor, msgLine),
-            type: match[2],
+            type,
             text: match[3],
             filePath,
           });