diff --git a/states.h b/states.h
--- a/states.h
+++ b/states.h
@@ -1,6 +1,8 @@
 #ifndef STATES_H
 #define STATES_H
 
+#include "config.h"
+
 typedef struct {
     int32_t temp;
     uint8_t temp_frac;
@@ -50,14 +52,24 @@ enum state {
     STATE_MAINTAIN,
 
     STATE_TC_ERROR,
+	STATE_RESET,
 };
 
 enum GOTO_MODE {
-	MODE_HEAT = 0,
+	#ifdef BOOTLOADER_SHORTCUT
+	MODE_BOOTLOADER,
+	#endif
+	MODE_HEAT,
 	MODE_SETUP,
 	MODE_RESET,
-	MODE_BOOTLOADER,
 	MODE_SIZE,
 };
 
+enum RESET_MODE {
+	RESET_RESET = 0,
+	RESET_BOOTLOADER,
+	RESET_EXIT,
+	RESET_SIZE,
+};
+
 #endif