diff --git a/queries/latex/images.scm b/queries/latex/images.scm
index 5ea1f3f9..85cdd47c 100644
--- a/queries/latex/images.scm
+++ b/queries/latex/images.scm
@@ -1,7 +1,15 @@
-(inline_formula) @image.content @image
+(inline_formula
+  (#set! image.ext "math.tex"))
+  @image.content @image
 
-(displayed_equation) @image.content @image
-(math_environment) @image.content @image
+(displayed_equation
+  (#set! image.ext "math.tex"))
+  @image.content @image
+
+((math_environment
+  (#set! image.ext "math.tex"))
+  @image.content @image
+  (#not-has-ancestor? @image "displayed_equation"))
 
 (graphics_include
   (_ (path) @image.src)