diff --git a/java/LogisticRegression/src/LogisticRegression.java b/java/LogisticRegression/src/LogisticRegression.java
index 8a13407..21e8f22 100644
--- a/java/LogisticRegression/src/LogisticRegression.java
+++ b/java/LogisticRegression/src/LogisticRegression.java
@@ -109,7 +109,7 @@ private static void test_lr() {
 			{0, 0, 1, 1, 1, 0}
 		};
 		
-		double[][] test_Y = new double[test_N][n_in];
+		double[][] test_Y = new double[test_N][n_out];
 		
 		
 		// test
diff --git a/scala/LogisticRegression.scala b/scala/LogisticRegression.scala
index 2386f5e..bf9b229 100644
--- a/scala/LogisticRegression.scala
+++ b/scala/LogisticRegression.scala
@@ -113,7 +113,7 @@ object LogisticRegression {
       Array(0, 0, 1, 1, 1, 0)
     )
 
-    val test_Y: Array[Array[Double]] = Array.ofDim[Double](test_N, n_in)
+    val test_Y: Array[Array[Double]] = Array.ofDim[Double](test_N, n_out)
 
     // test
     var j: Int = 0