diff --git a/requirements-freqai.txt b/requirements-freqai.txt
index c3aa2e4db..060a5219d 100644
--- a/requirements-freqai.txt
+++ b/requirements-freqai.txt
@@ -3,7 +3,6 @@
 
 # Required for freqai
 scikit-learn==1.1.1
-scikit-optimize==0.9.0
 joblib==1.1.0
 catboost==1.0.4
 lightgbm==3.3.2
diff --git a/setup.py b/setup.py
index 7aa56bf81..33e35c5ab 100644
--- a/setup.py
+++ b/setup.py
@@ -12,6 +12,13 @@ hyperopt = [
     'progressbar2',
 ]
 
+freqai = [
+    'scikit-learn',
+    'joblib',
+    'catboost',
+    'lightgbm',
+]
+
 develop = [
     'coveralls',
     'flake8',
@@ -31,7 +38,7 @@ jupyter = [
     'nbconvert',
 ]
 
-all_extra = plot + develop + jupyter + hyperopt
+all_extra = plot + develop + jupyter + hyperopt + freqai
 
 setup(
     tests_require=[
@@ -79,6 +86,7 @@ setup(
         'plot': plot,
         'jupyter': jupyter,
         'hyperopt': hyperopt,
+        'freqai': freqai,
         'all': all_extra,
     },
 )