diff --git a/docs/source/_config.yml b/docs/source/_config.yml
index a705a71..6ca10c7 100644
--- a/docs/source/_config.yml
+++ b/docs/source/_config.yml
@@ -12,7 +12,8 @@ execute:
 
 parse:
   myst_enable_extensions:
-      - amsmath
+    - amsmath
+    - dollarmath
 
 launch_buttons:
   notebook_interface: "jupyterlab"
diff --git a/docs/source/primitives.md b/docs/source/primitives.md
index b8d3c34..575e4d1 100644
--- a/docs/source/primitives.md
+++ b/docs/source/primitives.md
@@ -27,7 +27,7 @@ NIR defines 16 fundamental primitives listed in the table below, which backends
 | **Scale**                          | $s$                                                                       | $s I$                                                    | -                                                                                      |
 | **SumPooling**                     | $p$                                                                       | $\sum_{j} x_j$                                           | -                                                                                      |
 | **AvgPooling**                     | $p$                                                                       | **SumPooling**; **Scale**                                | -                                                                                      |
-| **Spike**                          | $\theta_\text{thr}$                                                       | $\dirac(I - \theta_\text{thr})$                               | -                                                                                      |
+| **Spike**                          | $\theta_\text{thr}$                                                       | $\delta(I - \theta_\text{thr})$                               | -                                                                                      |
 
 
 Each primitive is defined by their own dynamical equation, specified in the [API docs](https://nnir.readthedocs.io/en/latest/).