diff --git a/LinearAdvectionPINN.ipynb b/LinearAdvectionPINN.ipynb
index 7d2ae00..582e0af 100644
--- a/LinearAdvectionPINN.ipynb
+++ b/LinearAdvectionPINN.ipynb
@@ -259,7 +259,7 @@
         "\r\n",
         "u = model([np.expand_dims(T.flatten(), axis=1), \r\n",
         "           np.expand_dims(X.flatten(), axis=1)])\r\n",
-        "u = np.reshape(u, (m, n))"
+        "u = np.reshape(u, (n, m))"
       ],
       "execution_count": 7,
       "outputs": []
@@ -347,4 +347,4 @@
       ]
     }
   ]
-}
\ No newline at end of file
+}