diff --git a/smithy.tmGrammar.json b/smithy.tmGrammar.json
index 3937097..af4d302 100644
--- a/smithy.tmGrammar.json
+++ b/smithy.tmGrammar.json
@@ -35,7 +35,7 @@
       ]
     },
     "traits": {
-      "match": "@([\\w]+)\\b",
+      "match": "@([\\w\\.#]+)\\b",
       "name": "support.function"
     },
     "camelcase": {