update extension

koka-lang · Aug 28, 2023 · 9d31d31 · 9d31d31
1 parent 3bfe022
commit 9d31d31
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/support/vscode/koka.language-koka/src/extension.ts b/support/vscode/koka.language-koka/src/extension.ts
@@ -309,10 +309,10 @@ class MainCodeLensProvider implements vscode.CodeLensProvider {
   public async provideCodeLenses(document: TextDocument, token: CancellationToken): Promise<CodeLens[] | undefined> {
 		const doc = document.getText()
     const main = doc.indexOf('\nfun main')
-    if (main < 0){
+    if (main <= 0){
       return []
     }
-		return [this.createCodeLens(document, main)]
+		return [this.createCodeLens(document, main+1)]
 	}
 
 	private createCodeLens(document: TextDocument, offset: number): CodeLens {