Index: firmware/App/Modes/TreatmentEnd.c
===================================================================
diff -u -r8000e5d68a7a0ed6d1b43d8e6778a246ff0d8e72 -r1942c708bc95f57e87b0bf6a472ebe824a3f38ed
--- firmware/App/Modes/TreatmentEnd.c	(.../TreatmentEnd.c)	(revision 8000e5d68a7a0ed6d1b43d8e6778a246ff0d8e72)
+++ firmware/App/Modes/TreatmentEnd.c	(.../TreatmentEnd.c)	(revision 1942c708bc95f57e87b0bf6a472ebe824a3f38ed)
@@ -17,14 +17,13 @@
 
 #include "AirTrap.h"
 #include "BloodFlow.h"
-#include "Buttons.h"
 #include "DialInFlow.h"
 #include "DialOutFlow.h"
 #include "ModeTreatment.h"
 #include "OperationModes.h"
 #include "SystemCommMessages.h"
 #include "TaskGeneral.h"
-#include "TreatmentStop.h"
+#include "TreatmentEnd.h"
 #include "Valves.h"
 
 /**
@@ -69,7 +68,7 @@
  * The initTreatmentEnd function initializes the Treatment End sub-mode
  * module.
  * @details Inputs: none
- * @details Outputs: Treatment Stop sub-mode module initialized.
+ * @details Outputs: Treatment End sub-mode module initialized.
  * @return none
  *************************************************************************/
 void initTreatmentEnd( void )