diff --git a/tree-sitter-pyf/vim-plugin/after/queries/haskell/injections.scm b/tree-sitter-pyf/vim-plugin/after/queries/haskell/injections.scm
index 3bfff2a..c32051a 100644
--- a/tree-sitter-pyf/vim-plugin/after/queries/haskell/injections.scm
+++ b/tree-sitter-pyf/vim-plugin/after/queries/haskell/injections.scm
@@ -1,5 +1,12 @@
 ; extends
 (quasiquote
-  ((quoter) @_quoter (#any-of? @_quoter "fmt" "fmtTrim"))
-  (quasiquote_body) @pyf
-)
+  (quoter) @_name
+  (#eq? @_name "fmt")
+  ((quasiquote_body) @injection.content)
+  (#set! injection.language "pyf"))
+
+(quasiquote
+  (quoter) @_name
+  (#eq? @_name "fmtTrim")
+  ((quasiquote_body) @injection.content)
+  (#set! injection.language "pyf"))