diff --git a/node/src/command.rs b/node/src/command.rs
index fd37c014..54bcbdd3 100644
--- a/node/src/command.rs
+++ b/node/src/command.rs
@@ -23,10 +23,7 @@ use crate::{
 	cli::{Cli, RelayChainCli, Subcommand},
 	cli_opt::{EthApi, RpcConfig},
 	parachain,
-	parachain::{
-		service::{self, dev, frontier_database_dir, krest, peaq, start_node},
-		ExtHostFunctions,
-	},
+	parachain::service::{self, dev, frontier_database_dir, krest, peaq, start_node},
 };
 
 trait IdentifyChain {
@@ -323,7 +320,9 @@ pub fn run() -> sc_cli::Result<()> {
 				match cmd {
 					BenchmarkCmd::Pallet(cmd) => {
 						with_runtime_or_err!(chain_spec, {
-							runner.sync_run(|config| cmd.run::<Block, ExtHostFunctions>(config))
+							runner.sync_run(|config| {
+								cmd.run::<Block, parachain::ExtHostFunctions>(config)
+							})
 						})
 					},
 					BenchmarkCmd::Block(cmd) => {
diff --git a/node/src/parachain/mod.rs b/node/src/parachain/mod.rs
index e0c7afd4..24885a48 100644
--- a/node/src/parachain/mod.rs
+++ b/node/src/parachain/mod.rs
@@ -14,7 +14,9 @@ pub mod dev_chain_spec;
 pub mod krest_chain_spec;
 pub mod peaq_chain_spec;
 
-pub use service::{build_import_queue, ExtHostFunctions};
+pub use service::build_import_queue;
+#[cfg(feature = "runtime-benchmarks")]
+pub use service::ExtHostFunctions;
 
 /// Node `ChainSpec` extensions.
 ///