Index: firmware/App/Modes/ModeTreatment.c
===================================================================
diff -u -rf73db01f3c0cb623b6e56da43ccfba2d922717c5 -rd9fd9d0557d5a158d01e8950316c58d0f7e8a25f
--- firmware/App/Modes/ModeTreatment.c	(.../ModeTreatment.c)	(revision f73db01f3c0cb623b6e56da43ccfba2d922717c5)
+++ firmware/App/Modes/ModeTreatment.c	(.../ModeTreatment.c)	(revision d9fd9d0557d5a158d01e8950316c58d0f7e8a25f)
@@ -740,11 +740,19 @@
     TREATMENT_STATE_T result = TREATMENT_DIALYSIS_STATE;
     U32 newTime = getMSTimerCount();
     U32 msSinceLast = calcTimeBetween( lastTreatmentTimeStamp, newTime );
+    DIALYSIS_STATE_T dialysisState = getDialysisState();
 
     // Update treatment time (unless delivering a saline bolus)
-    if ( getDialysisState() != DIALYSIS_SALINE_BOLUS_STATE )
+    if ( dialysisState != DIALYSIS_SALINE_BOLUS_STATE )
     {
+#ifdef DIALYZER_REPRIME_ENABLED
+        if ( dialysisState != DIALYSIS_DIALYZER_REPRIME_STATE )
+        {
+#endif
         treatmentTimeMS += msSinceLast;
+#ifdef DIALYZER_REPRIME_ENABLED
+        }
+#endif
     }
     lastTreatmentTimeStamp = newTime;