diff --git a/PINNs_NavierStokes_example.ipynb b/PINNs_NavierStokes_example.ipynb
index 48cf921..e833041 100644
--- a/PINNs_NavierStokes_example.ipynb
+++ b/PINNs_NavierStokes_example.ipynb
@@ -1132,8 +1132,8 @@
     "x_vort = data_vort['x'] \n",
     "y_vort = data_vort['y'] \n",
     "w_vort = data_vort['w'] \n",
-    "modes = np.asscalar(data_vort['modes'])\n",
-    "nel = np.asscalar(data_vort['nel'])    \n",
+    "modes = np.ndarray.item(data_vort['modes'])\n",
+    "nel = np.ndarray.item(data_vort['nel'])    \n",
     "\n",
     "xx_vort = np.reshape(x_vort, (modes+1,modes+1,nel), order = 'F')\n",
     "yy_vort = np.reshape(y_vort, (modes+1,modes+1,nel), order = 'F')\n",