diff --git a/gptc/model.py b/gptc/model.py
index 634d900..53e7bbc 100644
--- a/gptc/model.py
+++ b/gptc/model.py
@@ -62,7 +62,7 @@ class Model:
         }
         return probs
 
-    def get(self, token):
+    def get(self, token: str) -> Dict[str, float]:
         try:
             weights = self.weights[
                 gptc.tokenizer.hash_single(gptc.tokenizer.normalize(token))