diff --git a/Core/Src/main.c b/Core/Src/main.c
index 9a5a862..e574010 100644
--- a/Core/Src/main.c
+++ b/Core/Src/main.c
@@ -25,6 +25,7 @@
 #include "ADBMS_CMD_MAKROS.h"
 #include "can.h"
 #include "AMS_HighLevel.h"
+#include "state_machine.h"
 #include "TMP1075.h"
 #include "eeprom.h"
 #include "errors.h"
@@ -117,9 +118,10 @@ int main(void)
   MX_USART1_UART_Init();
   MX_TIM1_Init();
   /* USER CODE BEGIN 2 */
+  sm_init();
   tmp1075_init(&hi2c1);
   AMS_Init(&hspi1);
-  can_init(&hcan);
+  ams_can_init(&hcan);
 
   uint32_t target_time = HAL_GetTick() + 10000;
   /* USER CODE END 2 */
@@ -133,7 +135,7 @@ int main(void)
     /* USER CODE BEGIN 3 */
     AMS_Loop();
 
-    can_handle_send_status();
+    ams_can_send_status();
   }
   /* USER CODE END 3 */
 }