Index: firmware/App/Modes/ModeDrain.c
===================================================================
diff -u -rab7dfc361e5ee347141842b317f1fc2f16e4e3f3 -rf215311725c002c9b9f2915e3502b89ae6a93c91
--- firmware/App/Modes/ModeDrain.c	(.../ModeDrain.c)	(revision ab7dfc361e5ee347141842b317f1fc2f16e4e3f3)
+++ firmware/App/Modes/ModeDrain.c	(.../ModeDrain.c)	(revision f215311725c002c9b9f2915e3502b89ae6a93c91)
@@ -85,9 +85,9 @@
  * The transitionToDrainMode function prepares for transition to drain mode.
  * @details Inputs: none
  * @details Outputs: Drain mode initialized
- * @return none
+ * @return initial state
  *************************************************************************/
-void transitionToDrainMode( void )
+U32 transitionToDrainMode( void )
 {
     // re-initialize each time we transition to drain mode
     initDrainMode();
@@ -113,6 +113,8 @@
     setROPumpTargetFlowRate( TARGET_RO_FLOW_RATE_L, TARGET_RO_PRESSURE_PSI );
 #endif
     startHeater( DG_PRIMARY_HEATER );
+
+    return drainState;
 }
 
 /*********************************************************************//**