diff --git a/comfy/ldm/flux/layers.py b/comfy/ldm/flux/layers.py
index 1f2975fb1..3518a1922 100644
--- a/comfy/ldm/flux/layers.py
+++ b/comfy/ldm/flux/layers.py
@@ -6,6 +6,8 @@ from torch import Tensor, nn
 
 from .math import attention, rope
 
+# Fix import for some custom nodes, TODO: delete eventually.
+RMSNorm = None
 
 class EmbedND(nn.Module):
     def __init__(self, dim: int, theta: int, axes_dim: list):