Index: firmware/App/Modes/ModeDrain.c
===================================================================
diff -u -r92eb8fea402c19b6d250651f9e11b088c6d9efd1 -rbe5079c95b05c303878763b458dc0854a600317e
--- firmware/App/Modes/ModeDrain.c	(.../ModeDrain.c)	(revision 92eb8fea402c19b6d250651f9e11b088c6d9efd1)
+++ firmware/App/Modes/ModeDrain.c	(.../ModeDrain.c)	(revision be5079c95b05c303878763b458dc0854a600317e)
@@ -1,18 +1,19 @@
 /**************************************************************************
- *
- *  Copyright (c) 2019-2020 Diality Inc. - All Rights Reserved.
- *
- *  THIS CODE MAY NOT BE COPIED OR REPRODUCED IN ANY FORM, IN PART OR IN
- *  WHOLE, WITHOUT THE EXPLICIT PERMISSION OF THE COPYRIGHT OWNER.
- *
- *  @file ModeDrain.c
- *
- *  @date 20-Dec-2019
- *  @author L. Baloa
- *
- *  @brief Top-level state machine for the drain mode.
- *
- **************************************************************************/
+*
+* Copyright (c) 2019-2020 Diality Inc. - All Rights Reserved.
+*
+* THIS CODE MAY NOT BE COPIED OR REPRODUCED IN ANY FORM, IN PART OR IN
+* WHOLE, WITHOUT THE EXPLICIT PERMISSION OF THE COPYRIGHT OWNER.
+*
+* @file ModeDrain.c
+*
+* @author (last) Sean Nash
+* @date (last) 02-Jul-2020
+*
+* @author (original) Leonardo Baloa
+* @date (original) 20-Dec-2019
+*
+***************************************************************************/
 
 #include "DrainPump.h"
 #include "LoadCell.h"
@@ -115,7 +116,11 @@
     LOAD_CELL_ID_T drainWeightLoadCell = LOAD_CELL_A1;
 
     // determine which load cell to use for drain volume - we want weight of inactive reservoir
+#ifndef BETA_V1_BUILD
     if ( RESERVOIR_1 == getActiveReservoir() )
+#else
+    if ( RESERVOIR_2 == getActiveReservoir() )
+#endif
     {
         drainWeightLoadCell = LOAD_CELL_B1;
     }