diff --git a/datumbox-framework-core/src/main/java/com/datumbox/framework/core/machinelearning/modelselection/splitters/ShuffleSplitter.java b/datumbox-framework-core/src/main/java/com/datumbox/framework/core/machinelearning/modelselection/splitters/ShuffleSplitter.java
index <HASH>..<HASH> 100644
--- a/datumbox-framework-core/src/main/java/com/datumbox/framework/core/machinelearning/modelselection/splitters/ShuffleSplitter.java
+++ b/datumbox-framework-core/src/main/java/com/datumbox/framework/core/machinelearning/modelselection/splitters/ShuffleSplitter.java
@@ -59,7 +59,7 @@ public class ShuffleSplitter extends AbstractSplitter {
     }
 
     @Override
-    public Iterable<Split> split(Dataframe dataset) {
+    public Iterable<ShuffleSplitter.Split> split(Dataframe dataset) {
         final int n = dataset.size();
         if(proportion <=0.0 || proportion >=1.0) {
             throw new IllegalArgumentException("The train size should be between 0.0 and 1.0.");
Change return type on ShuffleSplitter.