Index: firmware/App/Modes/ModeTreatment.c
===================================================================
diff -u -r9f5e68247ff2f5214e8828a1b8152ea16941fe39 -rd16609f215cf5ba8e08b902cf12a9492b40ed2cc
--- firmware/App/Modes/ModeTreatment.c	(.../ModeTreatment.c)	(revision 9f5e68247ff2f5214e8828a1b8152ea16941fe39)
+++ firmware/App/Modes/ModeTreatment.c	(.../ModeTreatment.c)	(revision d16609f215cf5ba8e08b902cf12a9492b40ed2cc)
@@ -287,10 +287,6 @@
             // TODO - implement
             break;
 
-        case TREATMENT_DIALYSIS_END_STATE:
-            // TODO - implement
-            break;
-
         case TREATMENT_END_STATE:
             // TODO - implement
             endAirTrapControl();  // TODO - move to appropriate place
@@ -454,7 +450,7 @@
 
     // Check if we are in an appropriate treatment state for settings adjustment
     if ( ( MODE_TREA == currMode ) &&
-         ( currentTreatmentState > TREATMENT_START_STATE ) && ( currentTreatmentState < TREATMENT_DIALYSIS_END_STATE ) &&
+         ( currentTreatmentState > TREATMENT_START_STATE ) && ( currentTreatmentState < TREATMENT_END_STATE ) &&
          ( CALC_ELAPSED_TREAT_TIME_IN_MIN() < treatmentTime ) && ( treatmentTime >= MIN_TREATMENT_TIME_MINUTES ) )
     {
         F32 uFVolume;
@@ -494,7 +490,7 @@
             rejectReason = REQUEST_REJECT_REASON_NOT_IN_TREATMENT_MODE;
         }
         else if ( ( currentTreatmentState <= TREATMENT_START_STATE ) ||
-                  ( currentTreatmentState >= TREATMENT_DIALYSIS_END_STATE ) )
+                  ( currentTreatmentState >= TREATMENT_END_STATE ) )
         {
             rejectReason = REQUEST_REJECT_REASON_INVALID_TREATMENT_STATE;
         }
@@ -541,7 +537,7 @@
 
     // Check if we are in an appropriate treatment state for settings adjustment
     if ( ( MODE_TREA == currMode ) &&
-         ( currentTreatmentState > TREATMENT_START_STATE ) && ( currentTreatmentState < TREATMENT_DIALYSIS_END_STATE ) &&
+         ( currentTreatmentState > TREATMENT_START_STATE ) && ( currentTreatmentState < TREATMENT_END_STATE ) &&
          ( uFVolume <= MAX_UF_VOLUME_ML ) &&
          ( CALC_TREAT_TIME_REMAINING_IN_SECS() >= PREVENT_UF_VOL_CHANGE_IF_NEARLY_DONE_SEC ) )
     {
@@ -600,7 +596,7 @@
             rejectReason = REQUEST_REJECT_REASON_TREATMENT_TIME_OUT_OF_RANGE;
         }
         else if ( ( currentTreatmentState <= TREATMENT_START_STATE ) ||
-                  ( currentTreatmentState >= TREATMENT_DIALYSIS_END_STATE ) )
+                  ( currentTreatmentState >= TREATMENT_END_STATE ) )
         {
             rejectReason = REQUEST_REJECT_REASON_INVALID_TREATMENT_STATE;
         }
@@ -849,7 +845,7 @@
     {
         stopDialysis();
         elapsedTreatmentTimeInSecs = presTreatmentTimeSecs;
-        currentTreatmentState = TREATMENT_DIALYSIS_END_STATE;
+        currentTreatmentState = TREATMENT_END_STATE;
     }
     // Broadcast treatment time and state data at interval
     if ( ++treatmentTimeBroadcastTimerCtr >= TREATMENT_TIME_DATA_PUB_INTERVAL )