diff --git a/pom.xml b/pom.xml
index a330680c..02b684e3 100644
--- a/pom.xml
+++ b/pom.xml
@@ -32,7 +32,7 @@
         be accompanied by a bump in version number, regardless of how minor the change.
    0.10.1 -->
 
-    <version> 0.10.6</version>
+    <version> 0.10.7</version>
 
     <repositories>
         <repository>
diff --git a/src/main/java/quickml/supervised/predictiveModelOptimizer/PredictiveModelOptimizer.java b/src/main/java/quickml/supervised/predictiveModelOptimizer/PredictiveModelOptimizer.java
index 237c4675..e2dfba80 100644
--- a/src/main/java/quickml/supervised/predictiveModelOptimizer/PredictiveModelOptimizer.java
+++ b/src/main/java/quickml/supervised/predictiveModelOptimizer/PredictiveModelOptimizer.java
@@ -28,7 +28,7 @@ public class PredictiveModelOptimizer {
      * @param crossValidator - Model tester takes a configuration and returns the loss
      */
 
-    public PredictiveModelOptimizer(Map<String, ? extends FieldValueRecommender> fieldsToOptimize, SimpleCrossValidator crossValidator, int iterations) {
+    public PredictiveModelOptimizer(Map<String, ? extends FieldValueRecommender> fieldsToOptimize, CrossValidator crossValidator, int iterations) {
         this.fieldsToOptimize = fieldsToOptimize;
         this.crossValidator = crossValidator;
         this.iterations = iterations;