Index: firmware/App/Modes/ModePreTreat.c
===================================================================
diff -u -r0d9ab0530817e35acb17ad94985f9e099f91a90c -r926e2786c330484808d43585e512140e8545ce91
--- firmware/App/Modes/ModePreTreat.c	(.../ModePreTreat.c)	(revision 0d9ab0530817e35acb17ad94985f9e099f91a90c)
+++ firmware/App/Modes/ModePreTreat.c	(.../ModePreTreat.c)	(revision 926e2786c330484808d43585e512140e8545ce91)
@@ -713,6 +713,11 @@
     signalDialOutPumpHardStop();
     setDialInPumpTargetFlowRate( DIP_PATIENT_CONNECTION_FLOW_RATE_ML_MIN, MOTOR_DIR_FORWARD, PUMP_CONTROL_MODE_OPEN_LOOP );
     cmdStartDGTrimmerHeater();
+
+    if ( TRUE == getTestConfigStatus( TEST_CONFIG_EXPEDITE_PRE_TREATMENT ) )
+    {
+        setOcclusionInstallLevel();
+    }
 }
 
 /*********************************************************************//**
Index: firmware/App/Modes/ModeTreatment.c
===================================================================
diff -u -rc0cbf6f5a6ebaf362f840f2049669528628f9c76 -r926e2786c330484808d43585e512140e8545ce91
--- firmware/App/Modes/ModeTreatment.c	(.../ModeTreatment.c)	(revision c0cbf6f5a6ebaf362f840f2049669528628f9c76)
+++ firmware/App/Modes/ModeTreatment.c	(.../ModeTreatment.c)	(revision 926e2786c330484808d43585e512140e8545ce91)
@@ -221,11 +221,6 @@
     // Enable venous bubble detection in treatment mode
     setVenousBubbleDetectionEnabled( TRUE );
 
-    if ( TRUE == getTestConfigStatus( TEST_CONFIG_EXPEDITE_PRE_TREATMENT ) )
-    {
-        setOcclusionInstallLevel();
-    }
-
     return currentTreatmentState;
 }