diff --git a/lab3/Part1_IntroductionCapsa.ipynb b/lab3/Part1_IntroductionCapsa.ipynb
index e6871272..89494c8b 100644
--- a/lab3/Part1_IntroductionCapsa.ipynb
+++ b/lab3/Part1_IntroductionCapsa.ipynb
@@ -350,11 +350,11 @@
         "### Generate and visualize bias scores for data in test set ###\n",
         "\n",
         "# Call the risk-aware model to generate scores\n",
-        "predictions, bias = bias_wrapped_dense_NN(x_test)\n",
+        "predictions = bias_wrapped_dense_NN(x_test)\n",
         "\n",
         "# Visualize the relationship between the input data x and the bias\n",
         "fig, ax = plt.subplots(2, 1, figsize=(8,6))\n",
-        "ax[0].plot(x_test, bias, label='bias')\n",
+        "ax[0].plot(x_test, predictions.bias, label='bias')\n",
         "ax[0].set_ylabel('Estimated Bias')\n",
         "ax[0].legend()\n",
         "\n",