diff --git a/bindings/python/tree_sitter_dotenv/__init__.py b/bindings/python/tree_sitter_dotenv/__init__.py
index 9603df9..aafc566 100644
--- a/bindings/python/tree_sitter_dotenv/__init__.py
+++ b/bindings/python/tree_sitter_dotenv/__init__.py
@@ -12,10 +12,9 @@ def _get_query(name, file):
 
 
 def __getattr__(name):
-    # NOTE: uncomment these to include any queries that this grammar contains:
 
-    # if name == "HIGHLIGHTS_QUERY":
-    #     return _get_query("HIGHLIGHTS_QUERY", "highlights.scm")
+    if name == "HIGHLIGHTS_QUERY":
+        return _get_query("HIGHLIGHTS_QUERY", "highlights.scm")
     # if name == "INJECTIONS_QUERY":
     #     return _get_query("INJECTIONS_QUERY", "injections.scm")
     # if name == "LOCALS_QUERY":
@@ -28,7 +27,7 @@ def __getattr__(name):
 
 __all__ = [
     "language",
-    # "HIGHLIGHTS_QUERY",
+    "HIGHLIGHTS_QUERY",
     # "INJECTIONS_QUERY",
     # "LOCALS_QUERY",
     # "TAGS_QUERY",
diff --git a/bindings/python/tree_sitter_dotenv/__init__.pyi b/bindings/python/tree_sitter_dotenv/__init__.pyi
index abf6633..44a3085 100644
--- a/bindings/python/tree_sitter_dotenv/__init__.pyi
+++ b/bindings/python/tree_sitter_dotenv/__init__.pyi
@@ -1,8 +1,6 @@
 from typing import Final
 
-# NOTE: uncomment these to include any queries that this grammar contains:
-
-# HIGHLIGHTS_QUERY: Final[str]
+HIGHLIGHTS_QUERY: Final[str]
 # INJECTIONS_QUERY: Final[str]
 # LOCALS_QUERY: Final[str]
 # TAGS_QUERY: Final[str]