diff --git a/crates/apollo_state_reader/src/apollo_state.rs b/crates/apollo_state_reader/src/apollo_state.rs
index 73336578a18..ceb3aa0441a 100644
--- a/crates/apollo_state_reader/src/apollo_state.rs
+++ b/crates/apollo_state_reader/src/apollo_state.rs
@@ -142,6 +142,7 @@ impl ApolloReader {
class_hash: ClassHash,
) -> StateResult<(CasmContractClass, SierraContractClass)> {
let Some(class_reader) = &self.class_reader else {
+ // Class reader is not set. Try to read directly from storage.
let (option_casm, option_sierra) = self
.reader()?
.get_casm_and_sierra(&class_hash)
@@ -161,6 +162,7 @@ impl ApolloReader {
fn read_deprecated_casm(&self, class_hash: ClassHash) -> StateResult