diff --git a/usedMcode/ridgeCVvia.m b/usedMcode/ridgeCVvia.m
index 3e9790a..5b1e489 100644
--- a/usedMcode/ridgeCVvia.m
+++ b/usedMcode/ridgeCVvia.m
@@ -48,13 +48,16 @@
         predicted=predicted(1:end-eegOffset,:);
     end
 
-
-    while size(trainingData,1)>size(predicted,1)
-	trainingData=trainingData(2:end,:);
+    while size(trainingData,1)>size(viaData,1)
+        trainingData=trainingData(2:end,:);
     end
 
+    while size(trainingData,1)<size(viaData,1)
+        viaData=viaData(2:end,:);
+    end
+    
     while size(trainingData,1)<size(predicted,1)
-	predicted=predicted(2:end,:);
+        predicted=predicted(2:end,:);
     end
 
     viaCorr=zeros([k,size(viaData,2)]);