diff --git a/lua/lsp/config/pyright.lua b/lua/lsp/config/pyright.lua
index f5d491b..36c3837 100644
--- a/lua/lsp/config/pyright.lua
+++ b/lua/lsp/config/pyright.lua
@@ -12,13 +12,13 @@ return {
                 "--max-line-length=80",
                 "--max-doc-length=80",
                 "--format",
-                "%(row)d,%(col)d,%(code).1s: %(text)s",
+                "%(row)d,%(col)d,%(code)s: %(text)s",
                 "-",
             },
             stdin = true,
             stdout = true,
-            pattern = "^(%d+),(%d+),(%w): (.*)",
-            groups = { "lnum", "col", "severity", "message" },
+            pattern = "^(%d+),(%d+),((%w)%d+): (.*)",
+            groups = { "lnum", "col", "code", "severity", "message" },
             source = "flake8",
             severity_map = {
                 E = vim.diagnostic.severity.ERROR,