diff --git a/tee-worker/cli/src/trusted_base_cli/mod.rs b/tee-worker/cli/src/trusted_base_cli/mod.rs
index 7e880ed28a..75f36d7287 100644
--- a/tee-worker/cli/src/trusted_base_cli/mod.rs
+++ b/tee-worker/cli/src/trusted_base_cli/mod.rs
@@ -15,6 +15,8 @@
 
 */
 
+#[cfg(feature = "development")]
+use crate::trusted_base_cli::commands::litentry::remove_identity::RemoveIdentityCommand;
 use crate::{
 	trusted_base_cli::commands::{
 		balance::BalanceCommand,
@@ -22,7 +24,7 @@ use crate::{
 		get_shard_vault::GetShardVaultCommand,
 		litentry::{
 			get_storage::GetStorageCommand, link_identity::LinkIdentityCommand,
-			remove_identity::RemoveIdentityCommand, request_vc::RequestVcCommand,
+			request_vc::RequestVcCommand,
 			send_erroneous_parentchain_call::SendErroneousParentchainCallCommand,
 		},
 		nonce::NonceCommand,
diff --git a/tee-worker/litentry/pallets/identity-management/src/lib.rs b/tee-worker/litentry/pallets/identity-management/src/lib.rs
index 9db4213c73..65606fc841 100644
--- a/tee-worker/litentry/pallets/identity-management/src/lib.rs
+++ b/tee-worker/litentry/pallets/identity-management/src/lib.rs
@@ -221,7 +221,6 @@ pub mod pallet {
 			})
 		}
 
-		#[cfg(feature = "development")]
 		#[pallet::call_index(5)]
 		#[pallet::weight({15_000_000})]
 		pub fn remove_identity(