From 541bd2df22f92bcae3dd7fb35f011efedae30fa7 Mon Sep 17 00:00:00 2001
From: Xein <xeinmail@gmail.com>
Date: Sun, 1 Jan 2023 08:15:46 -0600
Subject: [PATCH] fix: pycode environment produced an error

---
 src/scanner.c                |  2 +-
 test/corpus/environments.txt | 22 ++++++++++++++++++++++
 2 files changed, 23 insertions(+), 1 deletion(-)

diff --git a/src/scanner.c b/src/scanner.c
index 57d6be0c2..c7fa5b10d 100644
--- a/src/scanner.c
+++ b/src/scanner.c
@@ -92,7 +92,7 @@ bool tree_sitter_latex_external_scanner_scan(void *payload, TSLexer *lexer,
                                              const bool *valid_symbols) {
   bool found = false;
   TSSymbol type = 0xFFFF;
-  for (int i = 0; i <= TRIVIA_RAW_ENV_MINTED; i++) {
+  for (int i = 0; i <= TRIVIA_RAW_ENV_PYCODE; i++) {
     if (valid_symbols[i]) {
       if (found) {
         return false;
diff --git a/test/corpus/environments.txt b/test/corpus/environments.txt
index 160cbdc95..4c787dd12 100644
--- a/test/corpus/environments.txt
+++ b/test/corpus/environments.txt
@@ -214,6 +214,28 @@ Minted environment
         (text
           (word))))))
 
+================================================================================
+Pycode environment
+================================================================================
+
+\begin{pycode}
+  print()
+\end{pycode}
+
+--------------------------------------------------------------------------------
+
+(source_file
+  (pycode_environment
+    (begin
+      (curly_group_text
+        (text
+          (word))))
+    (source_code)
+    (end
+      (curly_group_text
+        (text
+          (word))))))
+
 ================================================================================
 Math environment
 ================================================================================