diff --git a/site/en/tutorials/interpretability/integrated_gradients.ipynb b/site/en/tutorials/interpretability/integrated_gradients.ipynb
index b862696958e..6cf1e09b32a 100644
--- a/site/en/tutorials/interpretability/integrated_gradients.ipynb
+++ b/site/en/tutorials/interpretability/integrated_gradients.ipynb
@@ -904,8 +904,8 @@
         "    gradient_batch = one_batch(baseline, image, alpha_batch, target_class_idx)\n",
         "    gradient_batches.append(gradient_batch)\n",
         "      \n",
-        "  # Stack path gradients together row-wise into single tensor.\n",
-        "  total_gradients = tf.stack(gradient_batch)\n",
+        "  # Concatenate path gradients together row-wise into single tensor.\n",
+        "  total_gradients = tf.concat(gradient_batches, axis=0)\n",
         "\n",
         "  # Integral approximation through averaging gradients.\n",
         "  avg_gradients = integral_approximation(gradients=total_gradients)\n",
@@ -1151,7 +1151,6 @@
     "colab": {
       "collapsed_sections": [],
       "name": "integrated_gradients.ipynb",
-      "private_outputs": true,
       "toc_visible": true
     },
     "kernelspec": {