diff --git a/setup.py b/setup.py
index 7b693c6..c94a901 100755
--- a/setup.py
+++ b/setup.py
@@ -11,7 +11,7 @@
         'pandas>=0.23',
         'scipy>=1.1.0',
         'torch==1.0.1',
-        'molsets'
+        'molsets==0.1.3'
     ],
     description='Generative Tensorial Reinforcement Learning (GENTRL)',
 )