diff --git a/Exercise8/exercise8.ipynb b/Exercise8/exercise8.ipynb
index 6af1975..8cd6824 100755
--- a/Exercise8/exercise8.ipynb
+++ b/Exercise8/exercise8.ipynb
@@ -468,7 +468,7 @@
     "\n",
     "# From the matrix, we can compute statistics like average rating.\n",
     "print('Average rating for movie 1 (Toy Story): %f / 5' %\n",
-    "      np.mean(Y[0, R[0, :]]))\n",
+    "      np.mean(Y[0, R[0, :] == 1]))\n",
     "\n",
     "# We can \"visualize\" the ratings matrix by plotting it with imshow\n",
     "pyplot.figure(figsize=(8, 8))\n",