Index: firmware/App/Modes/Prime.c
===================================================================
diff -u -r72551346728009770b22d327a1ac4fad8b5a1675 -r174c9ba02da6790f01ea9141ab1cc1d28388f2f8
--- firmware/App/Modes/Prime.c	(.../Prime.c)	(revision 72551346728009770b22d327a1ac4fad8b5a1675)
+++ firmware/App/Modes/Prime.c	(.../Prime.c)	(revision 174c9ba02da6790f01ea9141ab1cc1d28388f2f8)
@@ -151,7 +151,7 @@
 
     setAlarmUserActionEnabled( ALARM_USER_ACTION_RESUME, TRUE );
     setAlarmUserActionEnabled( ALARM_USER_ACTION_RINSEBACK, FALSE );
-    setAlarmUserActionEnabled( ALARM_USER_ACTION_END_TREATMENT, FALSE );
+    setAlarmUserActionEnabled( ALARM_USER_ACTION_END_TREATMENT, TRUE );
 
     resetPrimeFlags();
 }
@@ -633,7 +633,7 @@
 {
     HD_PRE_TREATMENT_PRIME_STATE_T state = HD_PRIME_DIALYSATE_DIALYZER_STATE;
 
-    F32 const loadcellWeight = getReservoirWeightSmallFilter( DG_RESERVOIR_1 );
+    F32 const loadcellWeight = getLoadCellWeight( LOAD_CELL_RESERVOIR_1_PRIMARY );
     F32 const weightChange = fabs( 1.0 - ( previousLoadCellReading / loadcellWeight ) );
 
     if ( weightChange < LOAD_CELL_VOLUME_NOISE_TOLERANCE )
@@ -701,7 +701,7 @@
 static HD_PRE_TREATMENT_PRIME_STATE_T handlePrimeDialysateBypassState( void )
 {
     HD_PRE_TREATMENT_PRIME_STATE_T state = HD_PRIME_DIALYSATE_BYPASS_STATE;
-    F32 const loadcellWeight = getReservoirWeightSmallFilter( DG_RESERVOIR_2 );
+    F32 const loadcellWeight = getLoadCellWeight( LOAD_CELL_RESERVOIR_2_PRIMARY );
     F32 const weightChange = fabs( 1.0 - ( previousLoadCellReading / loadcellWeight ) );
 
     if ( weightChange < LOAD_CELL_VOLUME_NOISE_TOLERANCE )