Index: firmware/App/Modes/ModeDrain.c
===================================================================
diff -u -r2468e56fbecd26da713bc78535bd727f4b105fe1 -raf0faf02f1bd7bffcce083e9b52988a01c343d8e
--- firmware/App/Modes/ModeDrain.c	(.../ModeDrain.c)	(revision 2468e56fbecd26da713bc78535bd727f4b105fe1)
+++ firmware/App/Modes/ModeDrain.c	(.../ModeDrain.c)	(revision af0faf02f1bd7bffcce083e9b52988a01c343d8e)
@@ -74,9 +74,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();
@@ -102,6 +102,8 @@
     setROPumpTargetFlowRate( TARGET_RO_FLOW_RATE_L, TARGET_RO_PRESSURE_PSI );
 #endif
     startHeater( DG_PRIMARY_HEATER );
+
+    return drainState;
 }
 
 /*********************************************************************//**