Index: sources/view/hd/data/posttreatment/VPostTreatmentStatesData.cpp
===================================================================
diff -u -re2dc7bd9995a3bb410aa472a1f95c1cc9ba3136d -r2ef03b2ce51b4dc507f66e9671953a8e0824bde9
--- sources/view/hd/data/posttreatment/VPostTreatmentStatesData.cpp	(.../VPostTreatmentStatesData.cpp)	(revision e2dc7bd9995a3bb410aa472a1f95c1cc9ba3136d)
+++ sources/view/hd/data/posttreatment/VPostTreatmentStatesData.cpp	(.../VPostTreatmentStatesData.cpp)	(revision 2ef03b2ce51b4dc507f66e9671953a8e0824bde9)
@@ -1,15 +1,15 @@
 /*!
  *
- * Copyright (c) 2019-2020 Diality Inc. - All Rights Reserved.
+ * Copyright (c) 2021-2024 Diality Inc. - All Rights Reserved.
  * \copyright
  * THIS CODE MAY NOT BE COPIED OR REPRODUCED IN ANY FORM, IN PART OR IN
  * WHOLE, WITHOUT THE EXPLICIT PERMISSION OF THE COPYRIGHT OWNER.
  *
  * \file    VPostTreatmentStatesData.cpp
  * \author  (last)      Behrouz NematiPour
- * \date    (last)      01-Mar-2021
+ * \date    (last)      22-Mar-2023
  * \author  (original)  Behrouz NematiPour
- * \date    (original)  13-Aug-2020
+ * \date    (original)  11-Apr-2021
  *
  */
 #include "VPostTreatmentStatesData.h"
@@ -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 ***