diff --git a/usedMcode/ridgeCVvia.m b/usedMcode/ridgeCVvia.m
index 5b1e489..e82fe86 100644
--- a/usedMcode/ridgeCVvia.m
+++ b/usedMcode/ridgeCVvia.m
@@ -17,7 +17,6 @@
     
     factor=size(trainingDataEMG,1)/(size(trainingData,1)+eegOffset);
 
-
     if strcmp(via,'Autoenc')
         viaData=shiftingMean(synergiesAutoenc,factor);
     elseif strcmp(via,'PCA')
@@ -47,19 +46,20 @@
         viaData=viaData(1:end-eegOffset,:);
         predicted=predicted(1:end-eegOffset,:);
     end
-
-    while size(trainingData,1)>size(viaData,1)
+    
+    %make sure data are of same size
+    minSize=min(size(trainingData,1),size(viaData,1),size(predicted,1));
+    while minSize<size(trainingData,1)
         trainingData=trainingData(2:end,:);
     end
-
-    while size(trainingData,1)<size(viaData,1)
+    while minSize<size(viaData,1)
         viaData=viaData(2:end,:);
     end
-    
-    while size(trainingData,1)<size(predicted,1)
+    while minSize<size(predicted,1)
         predicted=predicted(2:end,:);
     end
 
+    
     viaCorr=zeros([k,size(viaData,2)]);
     finalCorr=zeros([k,size(predicted,2)]);