diff --git a/usedMcode/ridgeCVvia.m b/usedMcode/ridgeCVvia.m
index 4aa32ff..ec2b7cd 100644
--- a/usedMcode/ridgeCVvia.m
+++ b/usedMcode/ridgeCVvia.m
@@ -49,7 +49,7 @@
 
     viaDataPredicted=zeros(size(viaData));
     correlation=zeros(size(predicted,2));
-    viaCorrelation=zeros(size(predicted,2));
+    viaCorrelation=zeros(size(viaData,2));
 
     parfor j=1:size(viaData,2)
         pred=viaData(:,j);
@@ -57,7 +57,7 @@
         coeffs=ridge(pred,trainingData,ridgeParams(1),0);
         
         viaDataPredicted(:,j)=coeffs(1)+trainingData*coeffs(2:end);
-        viaCorrelation(:,j)=diag(corr(viaData(:,j),viaDataPredicted(:,j)));
+        viaCorrelation(j)=corr(pred,viaDataPredicted(:,j));
     end
     
     parfor j=1:size(predicted,2)
diff --git a/usedMcode/ridgeCorrelation.m b/usedMcode/ridgeCorrelation.m
index e2e5fec..8929a02 100644
--- a/usedMcode/ridgeCorrelation.m
+++ b/usedMcode/ridgeCorrelation.m
@@ -1,6 +1,6 @@
 function [correlation]=ridgeCorrelation(data,kin,coeffs)
     prediction=coeffs(1)+data*coeffs(2:end);
     
-    correlation=diag(corr(kin,prediction));
+    correlation=corr(kin,prediction);
 
 end
\ No newline at end of file