diff --git a/test/registered/kernels/ops/diffusion/test_model_fast_paths.py b/test/registered/kernels/ops/diffusion/test_model_fast_paths.py
index 6169e75b7978..718541d389ad 100644
--- a/test/registered/kernels/ops/diffusion/test_model_fast_paths.py
+++ b/test/registered/kernels/ops/diffusion/test_model_fast_paths.py
@@ -625,6 +625,7 @@ def _block(self):
         block.img_mod = nn.ModuleList([nn.Identity(), _CountingProjection(1.0)])
         block.txt_mod = nn.ModuleList([nn.Identity(), _CountingProjection(2.0)])
         block._modulation_cache = None
+        block._shard_modulation = False
         return block
 
     def _key(self, timestep, hidden, additional_t_cond=None):