diff --git a/main.py b/main.py
index 5f429d2..b9d37c2 100644
--- a/main.py
+++ b/main.py
@@ -37,10 +37,10 @@ def download_file_from_google_drive(file_id, output_path):
 # Add this directory to NLTK's data path
 nltk.data.path.append(nltk_data_dir)
 
-nltk.download('punkt') # At first you have to download these nltk packages.
-nltk.download('stopwords')
-nltk.download('wordnet')
-nltk.download('punkt_tab')
+nltk.download('punkt',download_dir=nltk_data_dir) # At first you have to download these nltk packages.
+nltk.download('stopwords',download_dir=nltk_data_dir)
+nltk.download('wordnet',download_dir=nltk_data_dir)
+nltk.download('punkt_tab',download_dir=nltk_data_dir)
 
 
 from nltk.corpus import stopwords