diff --git a/Rethinking_2/Chp_11.ipynb b/Rethinking_2/Chp_11.ipynb
index 8daaf12..664bf9e 100644
--- a/Rethinking_2/Chp_11.ipynb
+++ b/Rethinking_2/Chp_11.ipynb
@@ -7,8 +7,6 @@
    "outputs": [],
    "source": [
     "import warnings\n",
-    "\n",
-    "import aesara.tensor as at\n",
     "import arviz as az\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",
@@ -4719,7 +4717,7 @@
     "    s2 = 0.0 + b * income[2]  # pivoting the intercept for the third category\n",
     "    s = pm.math.stack([s0, s1, s2])\n",
     "\n",
-    "    p_ = at.nnet.softmax(s)\n",
+    "    p_ = pm.math.softmax(s, axis=1)\n",
     "    career_obs = pm.Categorical(\"career\", p=p_, observed=career)\n",
     "\n",
     "    trace_11_13 = pm.sample(tune=2000, target_accept=0.99, random_seed=RANDOM_SEED)\n",
@@ -5061,7 +5059,7 @@
     "    s2 = np.zeros(N)  # pivot\n",
     "    s = pm.math.stack([s0, s1, s2]).T\n",
     "\n",
-    "    p_ = at.nnet.softmax(s)\n",
+    "    p_ = pm.math.softmax(s, axis=1)\n",
     "    career_obs = pm.Categorical(\"career\", p=p_, observed=career)\n",
     "\n",
     "    trace_11_14 = pm.sample(1000, tune=2000, target_accept=0.9, random_seed=RANDOM_SEED)\n",