1 files changed, 8 insertions, 7 deletions
diff --git a/crates/rule-executor/src/state.rs b/crates/rule-executor/src/state.rs
index ec59519..efad4ea 100644
--- a/crates/rule-executor/src/state.rs
+++ b/crates/rule-executor/src/state.rs
@@ -5,9 +5,9 @@ use moka::future::Cache;
 use tokio::sync::RwLock;
 use tonic::transport::Endpoint;
 use tracing::error;
-use warden_core::configuration::{
-    routing::{RoutingConfiguration, query_routing_client::QueryRoutingClient},
-    rule::RuleConfigurationRequest,
+use warden_core::configuration::rule::{
+    RuleConfiguration, RuleConfigurationRequest,
+    query_rule_configuration_client::QueryRuleConfigurationClient,
 };
 use warden_stack::Configuration;
 
@@ -24,9 +24,9 @@ pub type AppHandle = Arc<AppState>;
 #[derive(Clone)]
 pub struct AppState {
     pub services: Services,
-    pub local_cache: Arc<RwLock<Cache<RuleConfigurationRequest, RoutingConfiguration>>>,
+    pub local_cache: Arc<RwLock<Cache<RuleConfigurationRequest, RuleConfiguration>>>,
     pub config: LocalConfig,
-    pub query_routing_client: QueryRoutingClient<Intercepted>,
+    pub query_rule_client: QueryRuleConfigurationClient<Intercepted>,
 }
 
 impl AppState {
@@ -42,13 +42,14 @@ impl AppState {
                 )
             })?;
 
-        let query_routing_client = QueryRoutingClient::with_interceptor(channel, MyInterceptor);
+        let query_rule_client =
+            QueryRuleConfigurationClient::with_interceptor(channel, MyInterceptor);
 
         Ok(Self {
             services,
             config,
             local_cache: Arc::new(RwLock::new(Cache::builder().build())),
-            query_routing_client,
+            query_rule_client,
         })
     }
 }