Index: firmware/App/Modes/TreatmentEnd.c
===================================================================
diff -u -r1a36e9bd718b40ceaee21dd12eca2da353828e30 -rf3d625e2593ced120c4a98b9ed7e28b132445642
--- firmware/App/Modes/TreatmentEnd.c	(.../TreatmentEnd.c)	(revision 1a36e9bd718b40ceaee21dd12eca2da353828e30)
+++ firmware/App/Modes/TreatmentEnd.c	(.../TreatmentEnd.c)	(revision f3d625e2593ced120c4a98b9ed7e28b132445642)
@@ -271,6 +271,19 @@
     {
         signalGoToRinseback();
     }
+    // set BP to air trap fill rate if air trap fill has been initiated
+    else if ( TRUE == isAirTrapFillInProgress() )
+    {
+        if ( 0 == getTargetBloodFlowRate() )
+        {
+            setBloodPumpTargetFlowRate( AIR_TRAP_FILL_BLOOD_FLOW_RATE, MOTOR_DIR_FORWARD, PUMP_CONTROL_MODE_CLOSED_LOOP );
+        }
+    }
+    // re-open VBV after air trap fill
+    else if ( VALVE_POSITION_C_CLOSE == getSetValvePosition( VBV ) )
+    {
+        setValvePosition( VBV, VALVE_POSITION_B_OPEN );
+    }
 
     return result;
 }