diff --git a/StockTrainer.py b/StockTrainer.py
index e9348e4..004b844 100644
--- a/StockTrainer.py
+++ b/StockTrainer.py
@@ -35,10 +35,7 @@ class StockTrainer:
                     temp_X[k][idx] = TOTAL_X[k][j]
                 idx += 1
             X.append(temp_X)
-            Y.append(TOTAL_Y[i])
-
-
-
+            Y.append(TOTAL_Y[0][i])
 
         return X, Y