Index: simulator/plugins/treatmentstates/loader.py
===================================================================
diff -u -rc13dc6ba2410716a3f1c446e564085aeb9114a60 -r69b2aacf31c5fdc28e078f87dbdee5c3ab672dd1
--- simulator/plugins/treatmentstates/loader.py	(.../loader.py)	(revision c13dc6ba2410716a3f1c446e564085aeb9114a60)
+++ simulator/plugins/treatmentstates/loader.py	(.../loader.py)	(revision 69b2aacf31c5fdc28e078f87dbdee5c3ab672dd1)
@@ -1,14 +1,15 @@
 """
     The Ultrafiltration ui loader
 """
-from runtimewidget import RunTimeWidget
+import os
+from simulator.runtimewidget import RunTimeWidget
 from PySide2 import QtWidgets
 from PySide2.QtCore import Slot
 from dialin.squish import denaliMessages
-from dialin.squish.denaliMessages import txStates, EResponse
+from dialin.squish.denaliMessages import txStates
 
 
-class TreatmentStates(RunTimeWidget):
+class Loader(RunTimeWidget):
     """
     The Ultrafiltration ui loader
     """
@@ -18,7 +19,7 @@
     tblHeparinStates: QtWidgets.QTableWidget
 
     def __init__(self):
-        super().__init__("plugins/treatmentstates/interface.ui")
+        super().__init__(os.path.dirname(__file__))
         self.initialize()
         self.init_widgets()
         self.init_connections()