diff --git a/queries/yang/indents.scm b/queries/yang/indents.scm
index d68a76529..aab7ddbad 100644
--- a/queries/yang/indents.scm
+++ b/queries/yang/indents.scm
@@ -1,7 +1,9 @@
 (module) @indent
 (submodule) @indent
 (statement) @indent
+(extension_statement) @indent
 (statement ";" @indent_end)
+(extension_statement ";" @indent_end)
 (block "}" @indent_end @branch)
 
 ((string) @aligned_indent