diff --git a/__init__.py b/__init__.py
index a740c7a..3d11af2 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,7 +1,7 @@
 from .nodes import *
 
 NODE_CLASS_MAPPINGS = {
-    "Measure Loss": MeasureLoss,
+    "Measure Timestep Loss": MeasureTimestepLoss,
 }
 
 __all__ = ['NODE_CLASS_MAPPINGS']
\ No newline at end of file
diff --git a/nodes.py b/nodes.py
index 524e20b..86ce5eb 100644
--- a/nodes.py
+++ b/nodes.py
@@ -37,7 +37,7 @@ def common_ksampler(model, seed, steps, scheduler, positive, latent, denoise=1.0
     loss = F.mse_loss(samples, latent_image, reduction="none").mean(dim=(1,2,3))
     return loss
 
-class MeasureLoss:
+class MeasureTimestepLoss:
     @classmethod
     def INPUT_TYPES(s):
         return {