diff --git a/docs/snntorch.functional.rst b/docs/snntorch.functional.rst
index 6b9f0a40..a25bcb0b 100644
--- a/docs/snntorch.functional.rst
+++ b/docs/snntorch.functional.rst
@@ -14,7 +14,7 @@ Example::
 
       net = Net().to(device)
       optimizer = torch.optim.Adam(net.parameters(), lr=lr, betas=betas)
-      criterion = SF.ce_count_loss()  # apply cross-entropy to spike count
+      loss_fn = SF.ce_count_loss()  # apply cross-entropy to spike count
 
       spk_rec, mem_rec = net(input_data)
       loss = loss_fn(spk_rec, targets)