diff --git a/contracts/whitelists/tiered-whitelist-flex/src/state.rs b/contracts/whitelists/tiered-whitelist-flex/src/state.rs
index 5ad0e29ed..3cdd99568 100644
--- a/contracts/whitelists/tiered-whitelist-flex/src/state.rs
+++ b/contracts/whitelists/tiered-whitelist-flex/src/state.rs
@@ -40,3 +40,5 @@ pub const ADMIN_LIST: Item<AdminList> = Item::new("admin_list");
 pub const CONFIG: Item<Config> = Item::new("config");
 
 pub const WHITELIST_STAGES: Map<(u32, Addr), u32> = Map::new("wl_stages");
+
+pub const MEMBER_COUNT: Map<u32, u32> = Map::new("member_count");
diff --git a/contracts/whitelists/tiered-whitelist-merkletree/src/msg.rs b/contracts/whitelists/tiered-whitelist-merkletree/src/msg.rs
index 79dbdbd92..96e37b3b6 100644
--- a/contracts/whitelists/tiered-whitelist-merkletree/src/msg.rs
+++ b/contracts/whitelists/tiered-whitelist-merkletree/src/msg.rs
@@ -138,12 +138,12 @@ pub struct CanExecuteResponse {
 
 #[cw_serde]
 pub struct StageResponse {
+    pub stage_id: u32,
     pub stage: Stage,
     pub merkle_root: String,
 }
 
 #[cw_serde]
 pub struct StagesResponse {
-    pub stages: Vec<Stage>,
-    pub merkle_roots: Vec<String>,
+    pub stages: Vec<StageResponse>,
 }