Index: firmware/App/Modes/ModeTreatment.c
===================================================================
diff -u -rf1e3338feb8fccadf01aa629aec4a55132f6fd77 -r2d40deb50ffb27667c3e57f427b1e197e163910e
--- firmware/App/Modes/ModeTreatment.c	(.../ModeTreatment.c)	(revision f1e3338feb8fccadf01aa629aec4a55132f6fd77)
+++ firmware/App/Modes/ModeTreatment.c	(.../ModeTreatment.c)	(revision 2d40deb50ffb27667c3e57f427b1e197e163910e)
@@ -477,19 +477,11 @@
  *************************************************************************/
 static TREATMENT_STATE_T handleTreatmentStartState( void )
 {
-//#ifndef TEST_UI_ONLY
     TREATMENT_STATE_T result = TREATMENT_BLOOD_PRIME_STATE;
-//#else
-//    TREATMENT_STATE_T result = TREATMENT_DIALYSIS_STATE;
-//#endif
 
     lastTreatmentTimeStamp = getMSTimerCount();
-//#ifndef TEST_UI_ONLY
     //transitionToBloodPrime(); // TODO
-//#else
-    setDialysisBloodPumpFlowRate( getTreatmentParameterU32( TREATMENT_PARAM_BLOOD_FLOW ) );
-    //transitionToDialysis();
-//#endif
+    setDialysisBloodPumpFlowRate( getTreatmentParameterU32( TREATMENT_PARAM_BLOOD_FLOW ) );  // TODO - remove later
 
     return result;
 }