diff --git a/asreview2-optuna/classifiers.py b/asreview2-optuna/classifiers.py
index 4a97baf..79d1956 100644
--- a/asreview2-optuna/classifiers.py
+++ b/asreview2-optuna/classifiers.py
@@ -18,7 +18,7 @@ def logistic_params(trial: optuna.trial.FrozenTrial):
     # Use logarithmic normal distribution for C (C effect is non-linear)
     C = trial.suggest_float("log__C", 0.01, 10, log=True)
 
-    solver = "lbfgs"
+    solver = trial.suggest_categorical("log__solver", ["lbfgs", "saga"])
 
     return {"C": C, "solver": solver}
 
diff --git a/asreview2-optuna/main.py b/asreview2-optuna/main.py
index f480f0e..8e3288e 100644
--- a/asreview2-optuna/main.py
+++ b/asreview2-optuna/main.py
@@ -21,7 +21,7 @@
 from feature_extractors import feature_extractor_params, feature_extractors
 
 # Study variables
-VERSION = 1
+VERSION = 2
 STUDY_SET = "demo"
 PICKLE_FOLDER_PATH = Path("synergy-dataset", "pickles")
 CLASSIFIER_TYPE = "log"  # Options: "nb", "log", "svm", "rf"