diff --git a/core/src/lib.rs b/core/src/lib.rs
index 150c945c4..6c067fa68 100644
--- a/core/src/lib.rs
+++ b/core/src/lib.rs
@@ -1,6 +1,6 @@
 use crate::{
     interfaces::{ProofRequest, RaikoError, RaikoResult},
-    preflight::{hekla, ontake},
+    preflight::preflight,
     provider::BlockDataProvider,
 };
 use alloy_primitives::Address;
@@ -17,7 +17,6 @@ use raiko_lib::{
     input::{GuestInput, GuestOutput, TaikoProverData},
     prover::{IdStore, IdWrite},
 };
-use reth_primitives::revm_primitives::SpecId;
 use reth_primitives::Header;
 use serde_json::Value;
 use std::{collections::HashMap, hint::black_box};
@@ -70,7 +69,7 @@ impl Raiko {
         //TODO: read fork from config
         let preflight_data = self.get_preflight_data();
         info!("Generating input for HEKLA fork");
-        hekla::preflight(provider, preflight_data)
+        preflight(provider, preflight_data)
             .await
             .map_err(Into::<RaikoError>::into)
     }
diff --git a/core/src/preflight.rs b/core/src/preflight.rs
index 4832dd83c..75a62295b 100644
--- a/core/src/preflight.rs
+++ b/core/src/preflight.rs
@@ -1,6 +1,5 @@
 use crate::{
     interfaces::{RaikoError, RaikoResult},
-    preflight::{block_time_to_block_slot, get_blob_data, get_headers},
     provider::{db::ProviderDb, rpc::RpcBlockDataProvider, BlockDataProvider},
     require,
 };