diff --git a/tensorflow_v2/notebooks/3_NeuralNetworks/neural_network_raw.ipynb b/tensorflow_v2/notebooks/3_NeuralNetworks/neural_network_raw.ipynb
index bbec2f13..5ac19d89 100644
--- a/tensorflow_v2/notebooks/3_NeuralNetworks/neural_network_raw.ipynb
+++ b/tensorflow_v2/notebooks/3_NeuralNetworks/neural_network_raw.ipynb
@@ -180,7 +180,7 @@
     "        loss = cross_entropy(pred, y)\n",
     "        \n",
     "    # Variables to update, i.e. trainable variables.\n",
-    "    trainable_variables = weights.values() + biases.values()\n",
+    "    trainable_variables = list(weights.values()) + list(biases.values())\n",
     "\n",
     "    # Compute gradients.\n",
     "    gradients = g.gradient(loss, trainable_variables)\n",