diff --git a/runtime/src/extensions.rs b/runtime/src/extensions.rs
index c3aec157b..46985c9d7 100644
--- a/runtime/src/extensions.rs
+++ b/runtime/src/extensions.rs
@@ -1,3 +1,4 @@
+use cumulus_pallet_parachain_system::RelaychainDataProvider;
 use cumulus_primitives_core::relay_chain::BlockNumber;
 use frame_support::{
 	dispatch::{GetDispatchInfo, RawOrigin},
@@ -13,7 +14,10 @@ use pop_api_primitives::{
 	CollectionId, ItemId,
 };
 use sp_core::crypto::UncheckedFrom;
-use sp_runtime::{traits::Dispatchable, DispatchError};
+use sp_runtime::{
+	traits::{BlockNumberProvider, Dispatchable},
+	DispatchError,
+};
 use sp_std::{boxed::Box, vec::Vec};
 use xcm::{
 	latest::{prelude::*, OriginKind::SovereignAccount},
@@ -210,7 +214,7 @@ where
 			ParachainSystemKeys::LastRelayChainBlockNumber => {
 				env.charge_weight(T::DbWeight::get().reads(1_u64))?;
 				let relay_block_num: BlockNumber =
-					cumulus_pallet_parachain_system::Pallet::<T>::last_relay_block_number();
+					RelaychainDataProvider::<T>::current_block_number();
 				log::debug!(
 					target:LOG_TARGET,
 					"{} last relay chain block number is: {:?}.", LOG_PREFIX, relay_block_num
@@ -608,7 +612,7 @@ mod tests {
 		new_test_ext().execute_with(|| {
 			let _ = env_logger::try_init();
 
-			let (wasm_binary, _) = load_wasm_module::<Runtime>("../contracts/pop-api-examples/read-runtime-state/target/ink/pop_api_extension_demo.wasm").unwrap();
+			let (wasm_binary, _) = load_wasm_module::<Runtime>("../contracts/pop-api-examples/read-runtime-state/target/ink/pop_api_read_state_example.wasm").unwrap();
 
 			let init_value = 100;