diff --git a/dist/whiteboxlayer-0.2.5-py3-none-any.whl b/dist/whiteboxlayer-0.2.5-py3-none-any.whl
new file mode 100644
index 0000000..ea1ad5c
Binary files /dev/null and b/dist/whiteboxlayer-0.2.5-py3-none-any.whl differ
diff --git a/setup.py b/setup.py
index c6c10cb..b820232 100644
--- a/setup.py
+++ b/setup.py
@@ -6,7 +6,7 @@
 
 setup(
     name                = 'whiteboxlayer',
-    version             = '0.2.4',
+    version             = '0.2.5',
     description         = 'TensorFlow based custom layers',
     author              = 'YeongHyeon Park',
     author_email        = 'young200405@gmail.com',
diff --git a/whiteboxlayer/extensions/recurrent.py b/whiteboxlayer/extensions/recurrent.py
index f3febab..cd410a2 100644
--- a/whiteboxlayer/extensions/recurrent.py
+++ b/whiteboxlayer/extensions/recurrent.py
@@ -45,14 +45,14 @@ def lstm_cell(layer, x_now, h_prev, c_prev, output_dim, \
     c_now_act = layer.activation(x=c_now, activation=activation, name="%s-c-act" %(name))
     h_now = tf.compat.v1.multiply(o_now, c_now_act, name="%s-h" %(name))
 
-    return h_now, c_now, y_now
+    return h_now, c_now
 
 def lstm_layer(layer, x, output_dim, \
     batch_norm=False, activation="tanh", recurrent_activation="sigmoid", bi_direction=False, name='lstm', verbose=True):
 
     x = tf.transpose(x, perm=[1, 0, 2])
     dim_seq = x.get_shape().as_list()[0]
-    y, y_rev, h_now = None, None, None
+    y, y_rev, h_now, c_now = None, None, None, None
     for idx_s in range(dim_seq):
 
         h_now, c_now = lstm_cell(layer=layer, \