Index: tests/test_uf.py
===================================================================
diff -u -rb93016ee67204223658152cb2952e8ab16af1b94 -r970faebcad339fa52192e9d6d578936f161b55a7
--- tests/test_uf.py	(.../test_uf.py)	(revision b93016ee67204223658152cb2952e8ab16af1b94)
+++ tests/test_uf.py	(.../test_uf.py)	(revision 970faebcad339fa52192e9d6d578936f161b55a7)
@@ -16,17 +16,24 @@
 
 import sys
 sys.path.append("..")
-from dialin.hd.hemodialysis_device import HD
-from dialin.dg.dialysate_generator import DG
-from dialin.hd.valves import ValvesEnum
-from dialin.dg.temperatures import DGTemperaturesNames
 from time import sleep
 
+from leahi_dialin.td.treatment_delivery import TD
+#from leahi_dialin.dd.dialysate_delivery import DD
+#from leahi_dialin.ro.reverse_osmosis import RO
+
+from leahi_dialin.td.modules.valves import ValvesEnum
+from leahi_dialin.td.modules.valves import ValvesPositions
+from leahi_dialin.td.modules.valves import ValvesStates
+from leahi_dialin.td.modules.valves import TDValves
+
 if __name__ == "__main__":
-    # create an HD object called hd
-    hd = HD(log_level="DEBUG")
-    # create a DG object called dg
-    dg = DG(log_level="DEBUG")
+    # create an TD object called td
+    td = TD(log_level="DEBUG")
+    # create a DD object called dd
+#    dd = DD(log_level="DEBUG")
+    # create a RO object called ro
+    #    ro = RO(log_level="DEBUG")
     sleep(2)
 
 #    hd.ui.cmd_ui_start_treatment_request()