diff --git a/crates/game-service/src/battle.rs b/crates/game-service/src/battle.rs
index 165d507..5733fc1 100644
--- a/crates/game-service/src/battle.rs
+++ b/crates/game-service/src/battle.rs
@@ -11,7 +11,7 @@ use logic::{table, LogicDataRef, LogicDataTables, LogicRandom};
 pub const LOCATION_FILE_COLUMN: &str = "FileName";
 pub const CHARACTER_KING_TOWER: &str = "KingTower";
 pub const CHARACTER_PRINCESS_TOWER: &str = "PrincessTower";
-pub const BUFF_TABLE: i32 = 9;
+pub const BUFF_ARRAY_TABLE: i32 = table::DAMAGE_TYPES;
 pub struct TowerSpec {
     pub data: LogicDataRef,
     pub instance: i32,
@@ -42,7 +42,7 @@ fn character_data(name: &str) -> LogicDataRef {
 impl BattleBuilder {
     pub fn new(root: impl AsRef<Path>) -> Self {
         let buff_type_count = LogicDataTables::instance()
-            .table(BUFF_TABLE)
+            .table(BUFF_ARRAY_TABLE)
             .map(|rows| rows.count())
             .unwrap_or(0);
         Self {