diff --git a/stock/util/Stock2Vector.py b/stock/util/Stock2Vector.py
index f5500c4..697d671 100644
--- a/stock/util/Stock2Vector.py
+++ b/stock/util/Stock2Vector.py
@@ -368,8 +368,8 @@ class Stock2Vector(HTS):
 
         size = len(label)
         batch_X, batch_Y = [], []
-        X = np.zeros((H_VECTOR_SIZE, W_VECTOR_SIZE))
         for i in range(W_VECTOR_SIZE - 1, size):
+            X = np.zeros((H_VECTOR_SIZE, W_VECTOR_SIZE))
             for w in range(i - W_VECTOR_SIZE + 1, i + 1):
                 X[0, w] = macd[w]
                 X[1, w] = diff_avg27[w]
@@ -404,8 +404,8 @@ class Stock2Vector(HTS):
                 X[30, w] = diff_open_lower[w]
                 X[31, w] = diff_close_upper[w]
 
-                batch_X.append(X)
-                batch_Y.append(label[i])
+            batch_X.append(X)
+            batch_Y.append(label[i])
 
         batch_X = np.asarray(batch_X)
         batch_Y = np.asarray(batch_Y, dtype='int64')