Update src/container.jl

TuringLang · Feb 3, 2022 · ed1afe1 · ed1afe1
1 parent 9088efc
commit ed1afe1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/container.jl b/src/container.jl
@@ -56,7 +56,7 @@ function forkr(trace::Trace)
     newf = reset_model(trace.f)
     Random123.set_counter!(trace.rng, 1)
 
-    ctask = Libtask.CTask(trace.ctask, trace.rng)
+    ctask = Libtask.CTask(newf, trace.rng)
 
     # add backward reference
     newtrace = Trace(newf, ctask, trace.rng)