Index: firmware/App/Modes/Dialysis.c
===================================================================
diff -u -r111d91eb16612b02663397a2faa713f2a9bafe0c -rdaefaef0d2d3e13ff563f11d9721ccc1d831cf06
--- firmware/App/Modes/Dialysis.c	(.../Dialysis.c)	(revision 111d91eb16612b02663397a2faa713f2a9bafe0c)
+++ firmware/App/Modes/Dialysis.c	(.../Dialysis.c)	(revision daefaef0d2d3e13ff563f11d9721ccc1d831cf06)
@@ -567,17 +567,9 @@
             currentUFState = handleUFRunningState( &result );
             break;
 
-        case UF_OFF_STATE:
-            currentUFState = handleUFOffState( &result );
-            break;
-
-        case UF_COMPLETED_STATE:
-            currentUFState = handleUFCompletedState( &result );
-            break;
-
         default:
             SET_ALARM_WITH_2_U32_DATA( ALARM_ID_HD_SOFTWARE_FAULT, SW_FAULT_ID_DIALYSIS_INVALID_UF_STATE, currentUFState )
-            currentUFState = UF_COMPLETED_STATE;
+            currentUFState = UF_PAUSED_STATE;
             break;
     }
 
@@ -637,16 +629,9 @@
 {
     UF_STATE_T result;
 
-    if ( maxUFVolumeML < NEARLY_ZERO )
-    {
-        result = UF_OFF_STATE;
-    }
-    else
-    {
-        lastUFTimeStamp = getMSTimerCount();
-        uFTimeMS = 0;
-        result = UF_RUNNING_STATE;
-    }
+    lastUFTimeStamp = getMSTimerCount();
+    uFTimeMS = 0;
+    result = UF_RUNNING_STATE;
 
     return result;
 }
@@ -725,7 +710,7 @@
     // If we have reached target UF volume, UF is complete
     if ( refUFVolume >= maxUFVolumeML )
     {
-        result = UF_COMPLETED_STATE;
+        setUFRate = 0.0;
     }
     // Handle saline bolus start request from user
     else if ( TRUE == salineBolusStartRequested )
@@ -750,74 +735,6 @@
 
 /*********************************************************************//**
  * @brief
- * The handleUFCompletedOrOffState function handles the UF Off state
- * of the ultrafiltration state machine.
- * @details Inputs: none
- * @details Outputs: UF volumes updated and provided to DPo pump controller.
- * @param dialysisState next dialysis state
- * @return next ultrafiltration state
- *************************************************************************/
-static UF_STATE_T handleUFOffState( DIALYSIS_STATE_T *dialysisState )
-{
-    UF_STATE_T result = UF_OFF_STATE;
-
-    // Calculate UF volumes and provide to dialysate outlet pump controller
-    updateUFVolumes();
-
-    // Handle saline bolus start request from user
-    if ( TRUE == salineBolusStartRequested )
-    {
-        salineBolusAutoResumeUF = FALSE;
-        // Go to saline bolus state
-        if ( SALINE_BOLUS_STATE_IDLE == currentSalineBolusState )
-        {
-            *dialysisState = DIALYSIS_SALINE_BOLUS_STATE;
-        }
-        else
-        {
-            salineBolusStartRequested = FALSE;
-        }
-    }
-
-    return result;
-}
-
-/*********************************************************************//**
- * @brief
- * The handleUFCompletedState function handles the UF Completed
- * state of the ultrafiltration state machine.  This is a terminal state.
- * @details Inputs: none
- * @details Outputs: UF volumes updated and provided to DPo pump controller.
- * @param dialysisState next dialysis state
- * @return next ultrafiltration state
- *************************************************************************/
-static UF_STATE_T handleUFCompletedState( DIALYSIS_STATE_T *dialysisState )
-{
-    UF_STATE_T result = UF_COMPLETED_STATE;
-
-    // Calculate UF volumes and provide to dialysate outlet pump controller
-    updateUFVolumes();
-
-    // Handle saline bolus start request from user
-    if ( TRUE == salineBolusStartRequested )
-    {
-        salineBolusAutoResumeUF = FALSE;
-        // Go to saline bolus state
-        if ( SALINE_BOLUS_STATE_IDLE == currentSalineBolusState )
-        {
-            *dialysisState = DIALYSIS_SALINE_BOLUS_STATE;
-        }
-        else
-        {
-            salineBolusStartRequested = FALSE;
-        }
-    }
-
-    return result;
-}
-
-/*********************************************************************//**
- * @brief
  * The handleSalineBolusIdleState function handles the idle state of the
  * saline bolus state machine.
  * @details Inputs: none