diff --git a/environment.yml b/environment.yml
index 713bd16..74f9309 100644
--- a/environment.yml
+++ b/environment.yml
@@ -11,3 +11,4 @@ dependencies:
   - pyradiomics
   - simpleitk
   - scikit-learn
+  - xgboost
diff --git a/requirements.txt b/requirements.txt
index 5097285..4917972 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,4 +4,5 @@ pandas
 pooch
 pyradiomics
 simpleitk
-sklearn
+scikit-learn
+xgboost