diff --git a/model.py b/model.py
index f30d55e..bd25e30 100644
--- a/model.py
+++ b/model.py
@@ -23,7 +23,7 @@ def get_cell():
       return cell_fn(args.rnn_size, state_is_tuple=False)
 
     cell = tf.contrib.rnn.MultiRNNCell(
-        [get_cell() for _ in range(args.rnn_size)])
+        [get_cell() for _ in range(args.num_layers)])
 
     if (infer == False and args.keep_prob < 1): # training mode
       cell = tf.contrib.rnn.DropoutWrapper(cell, output_keep_prob = args.keep_prob)