diff --git a/zeroshot_topics/zeroshot_tm.py b/zeroshot_topics/zeroshot_tm.py
index cd84cc0..7555c6c 100644
--- a/zeroshot_topics/zeroshot_tm.py
+++ b/zeroshot_topics/zeroshot_tm.py
@@ -1,6 +1,10 @@
 import attr
 from keybert import KeyBERT
 from .utils import load_zeroshot_model
+
+import nltk
+nltk.download('wordnet')
+
 from nltk.corpus import wordnet as wn
 
 classifier = load_zeroshot_model()