diff --git a/12_custom_models_and_training_with_tensorflow.ipynb b/12_custom_models_and_training_with_tensorflow.ipynb
index 2826b8a..60613bd 100644
--- a/12_custom_models_and_training_with_tensorflow.ipynb
+++ b/12_custom_models_and_training_with_tensorflow.ipynb
@@ -2890,7 +2890,7 @@
     "result = tf_cube(3)\n",
     "result = tf_cube(tf.constant([[1., 2.]])) # New shape: trace!\n",
     "result = tf_cube(tf.constant([[3., 4.], [5., 6.]])) # New shape: trace!\n",
-    "result = tf_cube(tf.constant([[7., 8.], [9., 10.], [11., 12.]])) # no trace"
+    "result = tf_cube(tf.constant([[7., 8.], [9., 10.], [11., 12.]])) # New shape: trace!\n"
    ]
   },
   {