Index: simulator/plugins/treatmentstates/loader.py
===================================================================
diff -u -r2204a65b0af94e7e555f0f4207fd2701c164d1d5 -r967521dff94783ce48b50abc8db6bcd67bb501bf
--- simulator/plugins/treatmentstates/loader.py	(.../loader.py)	(revision 2204a65b0af94e7e555f0f4207fd2701c164d1d5)
+++ simulator/plugins/treatmentstates/loader.py	(.../loader.py)	(revision 967521dff94783ce48b50abc8db6bcd67bb501bf)
@@ -9,7 +9,8 @@
 # parent
 from simulator.dynamicloader import DynamicLoader
 # hd Simulator
-from dialin.ui.hd_simulator import HDSimulator, TXStates
+from simulator.interface import SimulationInterface
+from dialin.ui.hd_simulator import TXStates
 # plugin specific
 # -- none --
 
@@ -31,8 +32,8 @@
     tblTreatmentEndStates: QtWidgets.QTableWidget
     tblTreatmentStopStates: QtWidgets.QTableWidget
 
-    def __init__(self, hd_simulator: HDSimulator):
-        super().__init__(os.path.dirname(__file__), hd_simulator)
+    def __init__(self, interface: SimulationInterface):
+        super().__init__(os.path.dirname(__file__), interface)
 
     def _init_loader(self):
         """
@@ -103,7 +104,7 @@
         blood_prime = int(self.tblBloodPrimeStates.verticalHeaderItem(self.tblBloodPrimeStates.currentRow()).text())
         treatment_end = int(self.tblTreatmentEndStates.verticalHeaderItem(self.tblTreatmentEndStates.currentRow()).text())
         treatment_stop = int(self.tblTreatmentStopStates.verticalHeaderItem(self.tblTreatmentStopStates.currentRow()).text())
-        self.hd_simulator.cmd_set_treatment_states_data(
+        self.interface.hd.cmd_set_treatment_states_data(
             sub_mode,
             uf_state,
             saline,