diff --git a/src/main/target/common_pre.h b/src/main/target/common_pre.h
index c3808c098d..f7346b0f9b 100644
--- a/src/main/target/common_pre.h
+++ b/src/main/target/common_pre.h
@@ -246,9 +246,12 @@ extern uint8_t _dmaram_end__;
 #define USE_SERIALRX_SPEKTRUM   // SRXL, DSM2 and DSMX protocol
 #define USE_SERIALRX_SUMD       // Graupner Hott protocol
 
-#if (TARGET_FLASH_SIZE > 128)
+#if (TARGET_FLASH_SIZE > 256)
+#define PID_PROFILE_COUNT 4
+#define CONTROL_RATE_PROFILE_COUNT  4
+#elif (TARGET_FLASH_SIZE > 128)
 #define PID_PROFILE_COUNT 3
-#define CONTROL_RATE_PROFILE_COUNT  6
+#define CONTROL_RATE_PROFILE_COUNT  4
 #else
 #define PID_PROFILE_COUNT 2
 #define CONTROL_RATE_PROFILE_COUNT  3
diff --git a/src/test/unit/platform.h b/src/test/unit/platform.h
index 2083af21cb..dec45fa176 100644
--- a/src/test/unit/platform.h
+++ b/src/test/unit/platform.h
@@ -32,8 +32,8 @@
 #define FAST_DATA_ZERO_INIT
 #define FAST_DATA
 
-#define PID_PROFILE_COUNT 3
-#define CONTROL_RATE_PROFILE_COUNT  6
+#define PID_PROFILE_COUNT 4
+#define CONTROL_RATE_PROFILE_COUNT  4
 #define USE_MAG
 #define USE_BARO
 #define USE_GPS