From 95afb70feca278c59ac364b34347a58ab42f4573 Mon Sep 17 00:00:00 2001
From: "Jens H. Nielsen" <Jens.Nielsen@microsoft.com>
Date: Fri, 21 Oct 2022 12:00:26 +0200
Subject: [PATCH] Add missing nodeName to two node classes

---
 plottr/node/filter/correct_offset.py | 3 +++
 plottr/node/scaleunits.py            | 1 +
 2 files changed, 4 insertions(+)

diff --git a/plottr/node/filter/correct_offset.py b/plottr/node/filter/correct_offset.py
index c2232936..189f26f2 100644
--- a/plottr/node/filter/correct_offset.py
+++ b/plottr/node/filter/correct_offset.py
@@ -36,6 +36,9 @@ def getAvgAxis(self) -> str:
 
 
 class SubtractAverage(Node):
+
+    nodeName = "SubtractAverage"
+
     useUi = True
     uiClass = SubtractAverageWidget
 
diff --git a/plottr/node/scaleunits.py b/plottr/node/scaleunits.py
index b5bb4f01..d04ee8b7 100644
--- a/plottr/node/scaleunits.py
+++ b/plottr/node/scaleunits.py
@@ -103,6 +103,7 @@ class ScaleUnits(Node):
     e.g (1*10**3 complexunit)
     """
     useUi = True
+    nodeName = "ScaleUnits"
     uiClass = ScaleUnitsWidget
 
     def __init__(self, name: str):