Index: firmware/App/Modes/ModePreTreat.c
===================================================================
diff -u -rd7cae63618be45748f93b1576ca49e7825fe0cf8 -rfdaecc8cba085b8f2a57c8478c6ad800f7303483
--- firmware/App/Modes/ModePreTreat.c	(.../ModePreTreat.c)	(revision d7cae63618be45748f93b1576ca49e7825fe0cf8)
+++ firmware/App/Modes/ModePreTreat.c	(.../ModePreTreat.c)	(revision fdaecc8cba085b8f2a57c8478c6ad800f7303483)
@@ -21,6 +21,7 @@
 #include "TaskGeneral.h"
 #include "Timers.h"
 #include "TxParams.h"
+#include "StatePreTxDrySelfTests.h"
 
 /**
  *  @addtogroup TDPreTreatmentMode
@@ -111,7 +112,7 @@
             break;
 
         case TD_PRE_TREATMENT_SELF_TEST_DRY_STATE:
-            currentPreTreatmentState = handleSelfTestDryState();
+            currentPreTreatmentState = handleDrySelfTestState();
             break;
 
         case TD_PRE_TREATMENT_PRIME_STATE:
@@ -141,6 +142,19 @@
 
 /*********************************************************************//**
  * @brief
+ * The getPreTreatmentSubState function gets the current
+ * pre-treatment mode state.
+ * @details Inputs: currentPreTreatmentState
+ * @details Outputs: none
+ * @return currentPreTreatmentState
+ *************************************************************************/
+TD_PRE_TREATMENT_MODE_STATE_T getPreTreatmentSubState( void )
+{
+    return currentPreTreatmentState;
+}
+
+/*********************************************************************//**
+ * @brief
  * The handleSelfTestDryState function performs dry self-test.
  * @details Inputs: none
  * @details Outputs: transition to prime state on user request