diff --git a/examples/walk_tree.py b/examples/walk_tree.py
index 247cf67..135283d 100644
--- a/examples/walk_tree.py
+++ b/examples/walk_tree.py
@@ -2,10 +2,10 @@
 from tree_sitter import Language, Parser, Tree, Node
 import tree_sitter_python
 
-PY_LANGUAGE = Language(tree_sitter_python.language(), "python")
+PY_LANGUAGE = Language(tree_sitter_python.language())
 
 parser = Parser()
-parser.set_language(PY_LANGUAGE)
+parser.language = PY_LANGUAGE
 
 tree = parser.parse(bytes("a = 1", "utf8"))