diff --git a/m1_tf_test.py b/m1_tf_test.py
index 306a038..063ef54 100644
--- a/m1_tf_test.py
+++ b/m1_tf_test.py
@@ -19,7 +19,7 @@ class ComputeSum(Layer):
 
   def __init__(self, input_dim):
       super(ComputeSum, self).__init__()
-      # Create a non-trainable weight.
+      # Create a trainable weight.
       self.total = tf.Variable(initial_value=tf.zeros((input_dim,)),
                                trainable=True)