From b1c569c13ddb189143a53d871654ece28e90be63 Mon Sep 17 00:00:00 2001
From: Ammar Arif <evergreenkary@gmail.com>
Date: Sat, 30 Nov 2024 14:44:09 -0500
Subject: [PATCH] fix tests

---
 crates/katana/storage/db/src/tables.rs | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/crates/katana/storage/db/src/tables.rs b/crates/katana/storage/db/src/tables.rs
index de5c1cf4aa..e4223d38cf 100644
--- a/crates/katana/storage/db/src/tables.rs
+++ b/crates/katana/storage/db/src/tables.rs
@@ -149,7 +149,6 @@ macro_rules! dupsort {
 }
 
 define_tables_enum! {[
-    (StageCheckpoints, TableType::Table),
     (Headers, TableType::Table),
     (BlockHashes, TableType::Table),
     (BlockNumbers, TableType::Table),
@@ -175,7 +174,8 @@ define_tables_enum! {[
     (StorageChangeSet, TableType::Table),
     (ClassTrie, TableType::Table),
     (ContractTrie, TableType::Table),
-    (ContractStorageTrie, TableType::Table)
+    (ContractStorageTrie, TableType::Table),
+    (StageCheckpoints, TableType::Table)
 ]}
 
 tables! {
@@ -280,6 +280,8 @@ mod tests {
         assert_eq!(Tables::ALL[22].name(), StorageChangeSet::NAME);
         assert_eq!(Tables::ALL[23].name(), ClassTrie::NAME);
         assert_eq!(Tables::ALL[24].name(), ContractTrie::NAME);
+        assert_eq!(Tables::ALL[25].name(), ContractStorageTrie::NAME);
+        assert_eq!(Tables::ALL[26].name(), StageCheckpoints::NAME);
 
         assert_eq!(Tables::Headers.table_type(), TableType::Table);
         assert_eq!(Tables::BlockHashes.table_type(), TableType::Table);
@@ -306,6 +308,8 @@ mod tests {
         assert_eq!(Tables::StorageChangeSet.table_type(), TableType::Table);
         assert_eq!(Tables::ClassTrie.table_type(), TableType::Table);
         assert_eq!(Tables::ContractTrie.table_type(), TableType::Table);
+        assert_eq!(Tables::ContractStorageTrie.table_type(), TableType::Table);
+        assert_eq!(Tables::StageCheckpoints.table_type(), TableType::Table);
     }
 
     use katana_primitives::address;