diff --git a/ext/DynamicPPLZygoteRulesExt.jl b/ext/DynamicPPLZygoteRulesExt.jl
index 93b87e2a0..78831fdc4 100644
--- a/ext/DynamicPPLZygoteRulesExt.jl
+++ b/ext/DynamicPPLZygoteRulesExt.jl
@@ -19,7 +19,7 @@ ZygoteRules.@adjoint function DynamicPPL.dot_observe(
         DynamicPPL.increment_num_produce!(vi)
         return sum(map(Distributions.loglikelihood, dists, value)), vi
     end
-    return ZygoteRules.pullback(__context__, dot_observe_fallback, spl, dists, value, vi)
+    return ZygoteRules.pullback(dot_observe_fallback, __context__, spl, dists, value, vi)
 end
 
 end # module