diff --git a/stock/util/Stock2Vector.py b/stock/util/Stock2Vector.py
index dc53279..0c0e180 100644
--- a/stock/util/Stock2Vector.py
+++ b/stock/util/Stock2Vector.py
@@ -368,12 +368,12 @@ class Stock2Vector(HTS):
 
         size = len(label)
         batch_X, batch_Y = [], []
-        CHANNEL_SIZE = 4
+        CHANNEL_SIZE = 3
         for i in range(VECTOR_SIZE*CHANNEL_SIZE-1, size):
             X = np.zeros((CHANNEL_SIZE, VECTOR_SIZE, VECTOR_SIZE))
             s = i - VECTOR_SIZE*CHANNEL_SIZE + 1
             e = s+VECTOR_SIZE
-            for c in range(0, 4):
+            for c in range(0, CHANNEL_SIZE):
                 if c > 0:
                     s = e
                     e += VECTOR_SIZE