diff --git a/setup.py b/setup.py
index 7d58563c..c5cc9e21 100644
--- a/setup.py
+++ b/setup.py
@@ -7,7 +7,7 @@
     "ipython"
 ]
 test_deps = [
-    'pytest>=4',
+    'pytest<=8.0.2',
     'pytest-cov>=2.6.0',
     'pytest-flake8',
     'flake8<5.0.0'
@@ -38,6 +38,7 @@
 
 vec_deps = [
     'gensim',
+    'smart-open==6.4.0',
     'transformers<4.29.0',
     "tensorboard",
     'torchvision',