Index: firmware/App/Modes/ModeDrain.c
===================================================================
diff -u -ra6e1a62344275a14e4edde1bdda9974eef840827 -ra5723df5741ceaf13cd6cf9cab50a55bfea15501
--- firmware/App/Modes/ModeDrain.c	(.../ModeDrain.c)	(revision a6e1a62344275a14e4edde1bdda9974eef840827)
+++ firmware/App/Modes/ModeDrain.c	(.../ModeDrain.c)	(revision a5723df5741ceaf13cd6cf9cab50a55bfea15501)
@@ -117,8 +117,9 @@
     // because the initial guess in the heaters driver needs the target flow to calculate
     // the new PWMs for the main and small primary heaters
 #ifndef DISABLE_FLOW_CONTROL_TREATMENT
-    setROPumpTargetFlowRate( TARGET_RO_FLOW_RATE_L, TARGET_RO_PRESSURE_PSI );
+    setROPumpTargetFlowRateLPM( TARGET_RO_FLOW_RATE_L, TARGET_RO_PRESSURE_PSI );
 #endif
+    setHeaterTargetTemperature( DG_PRIMARY_HEATER, getPrimaryHeaterTargetTemperature() );
     startHeater( DG_PRIMARY_HEATER );
 
     return drainState;