From 88e421b0c349084fec2fd97c7aa925a7188deff1 Mon Sep 17 00:00:00 2001
From: Brett Saviano <bsaviano@intersystems.com>
Date: Mon, 13 May 2024 11:43:17 -0400
Subject: [PATCH] #328: Improve fix

---
 server/src/providers/diagnostic.ts | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/server/src/providers/diagnostic.ts b/server/src/providers/diagnostic.ts
index 609d3d6..1a5b689 100644
--- a/server/src/providers/diagnostic.ts
+++ b/server/src/providers/diagnostic.ts
@@ -1101,7 +1101,13 @@ export async function onDiagnostics(params: DocumentDiagnosticParams): Promise<D
 						// Ran off the end of the implementation
 						(parsed[i][j].l == ld.cls_langindex) ||
 						// Ran off the end of the method
-						(/^objectscript(-int)?$/.test(doc.languageId) && j == 0 && parsed[i][j].l == ld.cos_langindex && parsed[i][j].s == ld.cos_label_attrindex) ||
+						(
+							// In a routine
+							/^objectscript(-int)?$/.test(doc.languageId) &&
+							// This is the first token and it's at the start of the line
+							j == 0 && parsed[i][j].p == 0 &&
+							// This token is a label
+							parsed[i][j].l == ld.cos_langindex && parsed[i][j].s == ld.cos_label_attrindex) ||
 						// Exited the script
 						(doc.languageId == "objectscript-csp" && parsed[i][j].l == ld.html_langindex)
 					) {