1 files changed, 5 insertions, 4 deletions
diff --git a/crates/vcs_actions/src/registry/client_registry.rs b/crates/vcs_actions/src/registry/client_registry.rs
index 5939bed..9769750 100644
--- a/crates/vcs_actions/src/registry/client_registry.rs
+++ b/crates/vcs_actions/src/registry/client_registry.rs
@@ -2,12 +2,13 @@ use action_system::{action::ActionContext, action_pool::ActionPool};
 use tcp_connection::error::TcpTargetError;
 
 use crate::{
-    actions::local_actions::register_hello_world_action, connection::protocol::RemoteActionInvoke,
+    actions::local_actions::register_set_upstream_vault_action,
+    connection::protocol::RemoteActionInvoke,
 };
 
 fn register_actions(pool: &mut ActionPool) {
     // Pool register here
-    register_hello_world_action(pool);
+    register_set_upstream_vault_action(pool);
 }
 
 pub fn client_action_pool() -> ActionPool {
@@ -25,11 +26,11 @@ pub fn client_action_pool() -> ActionPool {
 }
 
 async fn on_proc_begin(
-    ctx: &ActionContext,
+    ctx: &mut ActionContext,
     _args: &(dyn std::any::Any + Send + Sync),
 ) -> Result<(), TcpTargetError> {
     // Is ctx remote
-    let is_remote = ctx.is_remote();
+    let is_remote = ctx.is_remote_action();
 
     // Action name and arguments
     let action_name = ctx.action_name().to_string();