Index: firmware/App/Modes/TreatmentStop.c
===================================================================
diff -u -rd91a24c730aeb5cd7e3eba9ef4eca78e442911f8 -r37a8a58b766a496b39241dd7ae46dc10dbda35e4
--- firmware/App/Modes/TreatmentStop.c	(.../TreatmentStop.c)	(revision d91a24c730aeb5cd7e3eba9ef4eca78e442911f8)
+++ firmware/App/Modes/TreatmentStop.c	(.../TreatmentStop.c)	(revision 37a8a58b766a496b39241dd7ae46dc10dbda35e4)
@@ -15,6 +15,7 @@
 *
 ***************************************************************************/
 
+#include "AirTrap.h"
 #include "Buttons.h"
 #include "ModeTreatment.h"
 #include "OperationModes.h"
@@ -33,9 +34,8 @@
  * @brief
  * The initTreatmentStop function initializes the Treatment Stop sub-mode
  * module.
- * @details
- * Inputs : none
- * Outputs : Treatment Stop sub-mode module initialized.
+ * @details Inputs: none
+ * @details Outputs: Treatment Stop sub-mode module initialized.
  * @return none
  *************************************************************************/
 void initTreatmentStop( void )
@@ -46,22 +46,21 @@
  * @brief
  * The transitionToTreatmentStop function prepares for transition to treatment
  * stop sub-mode.
- * @details
- * Inputs : none
- * Outputs : none
+ * @details Inputs: none
+ * @details Outputs: none
  * @return none
  *************************************************************************/
 void transitionToTreatmentStop( void )
 {
+    endAirTrapControl();
 }
 
 /*********************************************************************//**
  * @brief
  * The execTreatmentStop function executes the Treatment Stop sub-mode
  * state machine.
- * @details
- * Inputs : none
- * Outputs : none
+ * @details Inputs: none
+ * @details Outputs: none
  * @return none
  *************************************************************************/
 void execTreatmentStop( void )