diff --git a/bin/katana/src/main.rs b/bin/katana/src/main.rs
index 8a2b524426..506c91d014 100644
--- a/bin/katana/src/main.rs
+++ b/bin/katana/src/main.rs
@@ -6,7 +6,8 @@ use clap::{CommandFactory, Parser};
use clap_complete::{generate, Shell};
use console::Style;
use katana_core::sequencer::KatanaSequencer;
-use katana_primitives::contract::{ClassHash, ContractAddress};
+use katana_primitives::class::ClassHash;
+use katana_primitives::contract::ContractAddress;
use katana_primitives::genesis::allocation::GenesisAccountAlloc;
use katana_primitives::genesis::Genesis;
use katana_rpc::{spawn, NodeHandle};
diff --git a/crates/katana/core/src/backend/contract.rs b/crates/katana/core/src/backend/contract.rs
index e66cf748ad..57966dc61c 100644
--- a/crates/katana/core/src/backend/contract.rs
+++ b/crates/katana/core/src/backend/contract.rs
@@ -1,4 +1,4 @@
-use katana_primitives::contract::DeprecatedCompiledClass;
+use katana_primitives::class::DeprecatedCompiledClass;
use starknet::core::types::FlattenedSierraClass;
pub enum StarknetContract {
diff --git a/crates/katana/core/src/sequencer.rs b/crates/katana/core/src/sequencer.rs
index ea3921e80f..7ed57474c2 100644
--- a/crates/katana/core/src/sequencer.rs
+++ b/crates/katana/core/src/sequencer.rs
@@ -12,9 +12,8 @@ use katana_executor::blockifier::utils::{block_context_from_envs, EntryPointCall
use katana_executor::blockifier::PendingState;
use katana_primitives::block::{BlockHash, BlockHashOrNumber, BlockIdOrTag, BlockNumber};
use katana_primitives::chain::ChainId;
-use katana_primitives::contract::{
- ClassHash, CompiledClass, ContractAddress, Nonce, StorageKey, StorageValue,
-};
+use katana_primitives::class::{ClassHash, CompiledClass};
+use katana_primitives::contract::{ContractAddress, Nonce, StorageKey, StorageValue};
use katana_primitives::event::{ContinuationToken, ContinuationTokenError};
use katana_primitives::receipt::Event;
use katana_primitives::transaction::{ExecutableTxWithHash, TxHash, TxWithHash};
diff --git a/crates/katana/executor/src/blockifier/state.rs b/crates/katana/executor/src/blockifier/state.rs
index 713b3da405..0e528b0874 100644
--- a/crates/katana/executor/src/blockifier/state.rs
+++ b/crates/katana/executor/src/blockifier/state.rs
@@ -3,7 +3,7 @@ use std::collections::HashMap;
use blockifier::state::cached_state::{CachedState, GlobalContractCache};
use blockifier::state::errors::StateError;
use blockifier::state::state_api::{StateReader, StateResult};
-use katana_primitives::contract::{CompiledClass, FlattenedSierraClass};
+use katana_primitives::class::{CompiledClass, FlattenedSierraClass};
use katana_primitives::conversion::blockifier::to_class;
use katana_primitives::FieldElement;
use katana_provider::traits::contract::ContractClassProvider;
@@ -16,9 +16,8 @@ use starknet_api::patricia_key;
use starknet_api::state::StorageKey;
mod primitives {
- pub use katana_primitives::contract::{
- ClassHash, CompiledClassHash, ContractAddress, Nonce, StorageKey, StorageValue,
- };
+ pub use katana_primitives::class::{ClassHash, CompiledClassHash};
+ pub use katana_primitives::contract::{ContractAddress, Nonce, StorageKey, StorageValue};
}
/// A state db only provide read access.
@@ -167,10 +166,7 @@ impl CachedStateWrapper {
}
impl ContractClassProvider for CachedStateWrapper {
- fn class(
- &self,
- hash: katana_primitives::contract::ClassHash,
- ) -> ProviderResult