diff --git a/queries/gomod/highlights.scm b/queries/gomod/highlights.scm
index ff33df3ab..b30be0944 100644
--- a/queries/gomod/highlights.scm
+++ b/queries/gomod/highlights.scm
@@ -14,6 +14,10 @@
 
 (module_path) @string.special.url
 
+(tool_directive) @keyword.directive
+
+(tool) @string.special.url
+
 [
   (version)
   (go_version)