diff --git a/ProgrammingAssignment_3/model.ipynb b/ProgrammingAssignment_3/model.ipynb
index 8fb20ef634f1e54dc5cb73d4414bc54055e146ea..e3645b70b007edab8bbc16eff177f28d825ee016 100644
--- a/ProgrammingAssignment_3/model.ipynb
+++ b/ProgrammingAssignment_3/model.ipynb
@@ -86,7 +86,7 @@
     "                fp += 1                            \n",
     "    \n",
     "    # returns the confusion matrix as numpy.ndarray\n",
-    "    return np.array([tp,fp, fn, tn])"
+    "    return np.array([[tp,fp], [fn, tn]])"
    ]
   }
  ],