Index: leahi_dialin/dd/dialysate_delivery.py
===================================================================
diff -u -rb3774ad1df35c613de1c09bef40a55eb0ce4af36 -r2dab2b0329a56006e07cd36a3883ed099d7a367a
--- leahi_dialin/dd/dialysate_delivery.py	(.../dialysate_delivery.py)	(revision b3774ad1df35c613de1c09bef40a55eb0ce4af36)
+++ leahi_dialin/dd/dialysate_delivery.py	(.../dialysate_delivery.py)	(revision 2dab2b0329a56006e07cd36a3883ed099d7a367a)
@@ -23,7 +23,9 @@
 from .modules.heaters import DDHeaters
 from .modules.levels import DDLevels
 from .modules.piston_pump import DDPistonPumps
+from .modules.post_gen_dialysate import DDPostGenDialysate
 from .modules.pressure_sensors import DDPressureSensors
+from .modules.pre_gen_dialysate import DDPreGenDialysate
 from .modules.temperature_sensors import DDTemperatureSensors
 from .modules.valves import DDValves
 
@@ -120,7 +122,9 @@
         self.heaters = DDHeaters(self.can_interface, self.logger)
         self.levels = DDLevels(self.can_interface, self.logger)
         self.piston_pumps = DDPistonPumps(self.can_interface, self.logger)
+        self.post_gen_dialysate = DDPostGenDialysate(self.can_interface, self.logger)
         self.pressure_sensors = DDPressureSensors(self.can_interface, self.logger)
+        self.pre_gen_dialysate = DDPreGenDialysate(self.can_interface, self.logger)
         self.temperature_sensors = DDTemperatureSensors(self.can_interface, self.logger)
         self.valves = DDValves(self.can_interface, self.logger)