diff --git a/Exercise2/exercise2.ipynb b/Exercise2/exercise2.ipynb
index b6e728b..1384883 100644
--- a/Exercise2/exercise2.ipynb
+++ b/Exercise2/exercise2.ipynb
@@ -422,7 +422,7 @@
     "    grad = np.zeros(theta.shape)\n",
     "\n",
     "    # ====================== YOUR CODE HERE ======================\n",
-    "    h = sigmoid(X.dot(theta.T))\n",
+    "    h = sigmoid(X.dot(theta))\n",
     "    \n",
     "    J = (1 / m) * np.sum(-y.dot(np.log(h)) - (1 - y).dot(np.log(1 - h)))\n",
     "    grad = (1 / m) * (h - y).dot(X)\n",