diff --git a/docs/usage/conservation.ipynb b/docs/usage/conservation.ipynb
index f4a29ba0..cdca01e2 100644
--- a/docs/usage/conservation.ipynb
+++ b/docs/usage/conservation.ipynb
@@ -286,8 +286,8 @@
     "interaction = transition.interactions[node_id]\n",
     "\n",
     "spin_magnitude_conservation(\n",
-    "    ingoing_spins=[incoming_state.particle.spin],\n",
-    "    outgoing_spins=[outgoing_state1.particle.spin, outgoing_state2.particle.spin],\n",
+    "    ingoing_spin_magnitudes=[incoming_state.particle.spin],\n",
+    "    outgoing_spin_magnitudes=[outgoing_state1.particle.spin, outgoing_state2.particle.spin],\n",
     "    interaction_qns=interaction,\n",
     ")"
    ]