diff --git a/Core/Src/main.c b/Core/Src/main.c
index 5b43ba9..532a88d 100644
--- a/Core/Src/main.c
+++ b/Core/Src/main.c
@@ -27,7 +27,8 @@
 
 /* Private includes ----------------------------------------------------------*/
 /* USER CODE BEGIN Includes */
-#ifdef BUILD_COMMON
+#include "common_cfg.h"
+#ifdef MEM_POOL_TOTAL_SIZE
 #include "common.h"
 #endif
 /* USER CODE END Includes */
@@ -107,7 +108,7 @@ int main(void)
 
   /* Infinite loop */
   /* USER CODE BEGIN WHILE */
-#ifdef BUILD_COMMON
+#ifdef MEM_POOL_TOTAL_SIZE
   RD_INIT();
   void Myprint_Init(void);
   Myprint_Init();