diff --git a/package.json b/package.json
index 700f055..23aabca 100644
--- a/package.json
+++ b/package.json
@@ -18,7 +18,7 @@
   },
   "tree-sitter": [
     {
-      "injection-regex": "^(nix-shell)$",
+      "injection-regex": "^(nix[-_]?shell)$",
       "scope": "source.nix_shell"
     }
   ]