Index: simulator/plugins/createtreatment/loader.py
===================================================================
diff -u -rdb5d7914d3c89371d7ce7e79dcd78626d1f06ee8 -r1779b6711f3791e8bf570e06ddb4c2e7d2a52e18
--- simulator/plugins/createtreatment/loader.py	(.../loader.py)	(revision db5d7914d3c89371d7ce7e79dcd78626d1f06ee8)
+++ simulator/plugins/createtreatment/loader.py	(.../loader.py)	(revision 1779b6711f3791e8bf570e06ddb4c2e7d2a52e18)
@@ -6,6 +6,7 @@
 from PySide2 import QtWidgets
 from PySide2.QtCore import Slot
 from dialin.squish import denaliMessages
+from dialin.ui.hd_simulator import HDSimulator
 
 
 class Loader(DynamicLoader):
@@ -24,8 +25,8 @@
     spnUFVolumeRejectReason: QtWidgets.QSpinBox
     sldUFVolume: QtWidgets.QSlider
 
-    def __init__(self):
-        super().__init__(os.path.dirname(__file__))
+    def __init__(self, hd_simulator: HDSimulator):
+        super().__init__(os.path.dirname(__file__), hd_simulator)
 
     def _init_loader(self):
         """