Index: sources/view/hd/data/posttreatment/VPostTreatmentStatesData.cpp
===================================================================
diff -u -rc9f8f8cf3c6c37fc6460d8675c62c9442c4d4263 -r5d142e2ec2aa4bdfd2567eb521c7208f76e16ae7
--- sources/view/hd/data/posttreatment/VPostTreatmentStatesData.cpp	(.../VPostTreatmentStatesData.cpp)	(revision c9f8f8cf3c6c37fc6460d8675c62c9442c4d4263)
+++ sources/view/hd/data/posttreatment/VPostTreatmentStatesData.cpp	(.../VPostTreatmentStatesData.cpp)	(revision 5d142e2ec2aa4bdfd2567eb521c7208f76e16ae7)
@@ -26,10 +26,12 @@
  */
 void View::VPostTreatmentStates::onActionReceive(const PostTreatmentStatesData &vData)
 {
-    subMode                 ( vData.mSubMode );
+    subMode                 ( vData.mSubMode    );
+    drainState              ( vData.mDrainState );
 
-    patientDisconnection    ( vData.mSubMode == GuiPostTreatmentStates::HD_POST_TREATMENT_PATIENT_DISCONNECTION_STATE   );
     drainReservoirs         ( vData.mSubMode == GuiPostTreatmentStates::HD_POST_TREATMENT_DRAIN_RESERVOIRS_STATE        );
+    patientDisconnection    ( vData.mSubMode == GuiPostTreatmentStates::HD_POST_TREATMENT_PATIENT_DISCONNECTION_STATE   );
+    disposableRemoval       ( vData.mSubMode == GuiPostTreatmentStates::HD_POST_TREATMENT_DISPOSABLE_REMOVAL_STATE      );
     verify                  ( vData.mSubMode == GuiPostTreatmentStates::HD_POST_TREATMENT_VERIFY_STATE                  );
 
     // DEBUG: *** Only for test ***