Index: unittests/tst_models.h
===================================================================
diff -u -r14c541d2b780b0a244bc84860aa565cf8dd7fe4b -rdf0c8b4fa51d4f5ef0216eb2c3f6f5dfc97bf12e
--- unittests/tst_models.h	(.../tst_models.h)	(revision 14c541d2b780b0a244bc84860aa565cf8dd7fe4b)
+++ unittests/tst_models.h	(.../tst_models.h)	(revision df0c8b4fa51d4f5ef0216eb2c3f6f5dfc97bf12e)
@@ -66,6 +66,14 @@
     void tst_MAdjustPressuresLimitsResponse_data ();
     void tst_MAdjustPressuresLimitsResponse      ();
 
+    // -- Treatment Saline Bolus
+    void tst_MSalineBolusResponse_data();
+    void tst_MSalineBolusResponse();
+
+    // -- Treatment Heparin
+    void tst_MHeparinResponse_data();
+    void tst_MHeparinResponse();
+
     // - Data Messages
     void tst_DGROPumpData_data();
     void tst_DGROPumpData();
@@ -99,6 +107,10 @@
     void tst_MTreatmentTimeData();
     void tst_MPressureOcclusionData_data();
     void tst_MPressureOcclusionData();
+    void tst_MTreatmentSalineBolusData_data();
+    void tst_MTreatmentSalineBolusData();
+    void tst_MTreatmentHeparinData_data();
+    void tst_MTreatmentHeparinData();
     void tst_MPowerOff_data();
     void tst_MPowerOff();
     void tst_MPrimingData_data();
@@ -122,10 +134,4 @@
     // Treatment Parameters
     void tst_MTreatmentParametersResp_data();
     void tst_MTreatmentParametersResp();
-
-    void tst_MTreatmentSalineBolusData_data();
-    void tst_MTreatmentSalineBolusData();
-
-    void tst_MSalineBolusResponse_data();
-    void tst_MSalineBolusResponse();
 };