diff --git a/test/consumers/rust/Cargo.toml b/test/consumers/rust/Cargo.toml
index 67ebdcd..aac5f5b 100644
--- a/test/consumers/rust/Cargo.toml
+++ b/test/consumers/rust/Cargo.toml
@@ -5,5 +5,6 @@ edition = "2021"
 
 [dependencies]
 tree-sitter-tlaplus = {path = "../../.."}
-tree-sitter = "0.22.5"
+tree-sitter-language = "0.1.0"
+tree-sitter = "0.23"
 
diff --git a/test/consumers/rust/src/main.rs b/test/consumers/rust/src/main.rs
index d4ca813..ae032c3 100644
--- a/test/consumers/rust/src/main.rs
+++ b/test/consumers/rust/src/main.rs
@@ -2,7 +2,9 @@ use tree_sitter::{Parser, Query, QueryCursor};
 
 fn main() {
     let mut parser = Parser::new();
-    parser.set_language(&tree_sitter_tlaplus::language()).expect("Error loading TLA+ grammar");
+    parser
+        .set_language(&tree_sitter_tlaplus::LANGUAGE.into())
+        .expect("Error loading TLA+ grammar");
     let source_code = r#"
         ---- MODULE Test ----
         op ≜ ∀ n ∈ ℕ : n ≥ 0
@@ -10,10 +12,13 @@ fn main() {
     let tree = parser.parse(source_code, None).unwrap();
     println!("{}", tree.root_node().to_sexp());
 
-    let query = Query::new(&tree_sitter_tlaplus::language(), "(def_eq \"≜\") @capture").unwrap();
+    let query = Query::new(
+        &tree_sitter_tlaplus::LANGUAGE.into(),
+        "(def_eq \"≜\") @capture",
+    )
+    .unwrap();
     let mut cursor = QueryCursor::new();
     for capture in cursor.matches(&query, tree.root_node(), "".as_bytes()) {
         println!("{:?}", capture);
     }
 }
-