diff --git a/setup.py b/setup.py
index 51268f9..216e845 100644
--- a/setup.py
+++ b/setup.py
@@ -32,6 +32,7 @@
     "lspopt",
     "ipywidgets",
     "joblib",
+    "lightgbm",
 ]
 
 PACKAGES = [
diff --git a/yasa/staging.py b/yasa/staging.py
index a0e1efb..9a59ffa 100644
--- a/yasa/staging.py
+++ b/yasa/staging.py
@@ -23,7 +23,7 @@ class SleepStaging:
     """
     Automatic sleep staging of polysomnography data.
 
-    To run the automatic sleep staging, you must install the
+    The automatic sleep staging requires the
     `LightGBM <https://lightgbm.readthedocs.io/>`_ and
     `antropy <https://github.com/raphaelvallat/antropy>`_ packages.