diff --git a/test/test_JuliaBUGS.jl b/test/test_JuliaBUGS.jl
index c7df6fed4..bc62433ee 100644
--- a/test/test_JuliaBUGS.jl
+++ b/test/test_JuliaBUGS.jl
@@ -55,7 +55,7 @@ end
 end
 
 @testset "Invariance test" begin
-    res = Pigeons.invariance_test(target, SliceSampler(), rng; condition_on=(:n_heads,)) 
+    res = Pigeons.invariance_test(unid_target, SliceSampler(), rng; condition_on=(:n_heads,)) 
     @show res.pvalues
     @test res.passed
 end