diff --git a/test/debug_utils.jl b/test/debug_utils.jl
index 0152282b7..a7a9f7b71 100644
--- a/test/debug_utils.jl
+++ b/test/debug_utils.jl
@@ -3,7 +3,7 @@
         # HACK: Require a model to instantiate it, so let's just grab one.
         model = first(DynamicPPL.TestUtils.DEMO_MODELS)
         context = DynamicPPL.DebugUtils.DebugContext(model)
-        DynamicPPL.TestUtils.test_context(context)
+        DynamicPPL.TestUtils.test_context(context, model)
     end
 
     @testset "$(model.f)" for model in DynamicPPL.TestUtils.DEMO_MODELS