diff --git a/evaluate.py b/evaluate.py
index a55ee4e..f97a026 100644
--- a/evaluate.py
+++ b/evaluate.py
@@ -135,7 +135,7 @@ def evaluate(args):
 
     parser.add_argument('--static', help='Which static network to use.', required=True)
     parser.add_argument('--flow', help='Which optical flow method to use.', required=True)
-    parser.add_argument('--frames', dest='frames', type=int, help='Number of frames to use.', default=5, required=False)
+    parser.add_argument('--frames', type=int, help='Number of frames to use.', default=5, required=False)
 
     args = parser.parse_args()
 
diff --git a/models/stgru.py b/models/stgru.py
index 5434aba..ca982ca 100644
--- a/models/stgru.py
+++ b/models/stgru.py
@@ -78,7 +78,7 @@ def get_GRU_cell(self, input_image, prev_image, flow_input, h_prev, unary_input)
         return h
 
     def softmax_last_dim(self, x):
-        # apply softmax to a 4D vector along the last dimension
+        # apply softmax to a 4D tensor along the last dimension
         S = tf.shape(x)
         y = tf.reshape(x, [-1, S[4-1]])
         y = tf.nn.softmax(y)