diff --git a/package.json b/package.json
index 259dfeb..57c8b59 100644
--- a/package.json
+++ b/package.json
@@ -21,7 +21,8 @@
       "first-line-regex": "^#!.*\\bnix-shell\\b.*$",
       "content-regex": "^\b$nix-shell wraps other languages so intentionally fail against (and yield to) those other language matches",
       "injection-regex": "^(nix[-_]?shell)$",
-      "scope": "source.nix_shell"
+      "scope": "source.nix_shell",
+      "highlights": [ "queries/highlights.scm" ]
     }
   ]
 }
diff --git a/queries/highlights.scm b/queries/highlights.scm
new file mode 100644
index 0000000..118d38e
--- /dev/null
+++ b/queries/highlights.scm
@@ -0,0 +1 @@
+(nix_shell_directive) @keyword