diff --git a/setup.py b/setup.py
index 91d5043..394d98a 100644
--- a/setup.py
+++ b/setup.py
@@ -21,7 +21,7 @@
 
 setup_requirements = []
 
-test_requirements = []
+test_requirements = ['lightgbm']
 
 setup(
     author="Kapil Sachdeva",