diff --git a/docs/src/examples/normalizing_flows.md b/docs/src/examples/normalizing_flows.md
index ac5c7af0e..6d4898ae3 100644
--- a/docs/src/examples/normalizing_flows.md
+++ b/docs/src/examples/normalizing_flows.md
@@ -29,7 +29,7 @@ function loss(θ)
 end
 
 function cb(p, l)
-    @info "FFJORD Training" loss=loss(p)
+    @info "FFJORD Training" loss=l
     return false
 end