diff --git a/NRS/nodes_NRS.py b/NRS/nodes_NRS.py
index 5986995..26da4bc 100644
--- a/NRS/nodes_NRS.py
+++ b/NRS/nodes_NRS.py
@@ -13,6 +13,7 @@ class PredictionType(Enum):
 _RAW_TO_ENUM = {
     "eps":          PredictionType.EPS,
     "epsilon":      PredictionType.EPS,
+    "flux":         PredictionType.EPS,
     "v":            PredictionType.V,
     "v_prediction": PredictionType.V,
     "x0":           PredictionType.X0,
@@ -218,4 +219,4 @@ class NRS:
 
 NODE_CLASS_MAPPINGS = {
     "NRS": NRS,
-}
\ No newline at end of file
+}