diff --git a/comfy/ldm/modules/diffusionmodules/mmdit.py b/comfy/ldm/modules/diffusionmodules/mmdit.py
index 94a38726e..aecf1ec03 100644
--- a/comfy/ldm/modules/diffusionmodules/mmdit.py
+++ b/comfy/ldm/modules/diffusionmodules/mmdit.py
@@ -858,7 +858,7 @@ class MMDiT(nn.Module):
             context = torch.cat(
                 (
                     repeat(self.register, "1 ... -> b ...", b=x.shape[0]),
-                    torch.Tensor([]).type_as(x) is context is None else context,
+                    torch.Tensor([]).type_as(x) if context is None else context,
                 ),
                 1,
             )