diff --git a/src/context_implementations.jl b/src/context_implementations.jl
index 515dce7ae..9fcd2e310 100644
--- a/src/context_implementations.jl
+++ b/src/context_implementations.jl
@@ -63,13 +63,17 @@ function tilde_assume(::IsParent, context::AbstractContext, args...)
     return tilde_assume(childcontext(context), args...)
 end
 
-function tilde_assume(rng, context::AbstractContext, args...)
+function tilde_assume(rng::Random.AbstractRNG, context::AbstractContext, args...)
     return tilde_assume(NodeTrait(tilde_assume, context), rng, context, args...)
 end
-function tilde_assume(::IsLeaf, rng, context::AbstractContext, sampler, right, vn, vi)
+function tilde_assume(
+    ::IsLeaf, rng::Random.AbstractRNG, context::AbstractContext, sampler, right, vn, vi
+)
     return assume(rng, sampler, right, vn, vi)
 end
-function tilde_assume(::IsParent, rng, context::AbstractContext, args...)
+function tilde_assume(
+    ::IsParent, rng::Random.AbstractRNG, context::AbstractContext, args...
+)
     return tilde_assume(rng, childcontext(context), args...)
 end
 
@@ -121,7 +125,9 @@ end
 function tilde_assume(context::PrefixContext, right, vn, vi)
     return tilde_assume(context.context, right, prefix(context, vn), vi)
 end
-function tilde_assume(rng, context::PrefixContext, sampler, right, vn, vi)
+function tilde_assume(
+    rng::Random.AbstractRNG, context::PrefixContext, sampler, right, vn, vi
+)
     return tilde_assume(rng, context.context, sampler, right, prefix(context, vn), vi)
 end
 
@@ -291,7 +297,7 @@ end
 function dot_tilde_assume(context::AbstractContext, args...)
     return dot_tilde_assume(NodeTrait(dot_tilde_assume, context), context, args...)
 end
-function dot_tilde_assume(rng, context::AbstractContext, args...)
+function dot_tilde_assume(rng::Random.AbstractRNG, context::AbstractContext, args...)
     return dot_tilde_assume(NodeTrait(dot_tilde_assume, context), rng, context, args...)
 end
 
@@ -309,7 +315,9 @@ function dot_tilde_assume(::IsParent, rng, context::AbstractContext, args...)
     return dot_tilde_assume(rng, childcontext(context), args...)
 end
 
-function dot_tilde_assume(rng, ::DefaultContext, sampler, right, left, vns, vi)
+function dot_tilde_assume(
+    rng::Random.AbstractRNG, ::DefaultContext, sampler, right, left, vns, vi
+)
     return dot_assume(rng, sampler, right, vns, left, vi)
 end