1 files changed, 10 insertions, 7 deletions
diff --git a/crates/vcs_actions/src/registry/client_registry.rs b/crates/vcs_actions/src/registry/client_registry.rs
index 56acdad..47fd7ee 100644
--- a/crates/vcs_actions/src/registry/client_registry.rs
+++ b/crates/vcs_actions/src/registry/client_registry.rs
@@ -2,13 +2,12 @@ use action_system::{action::ActionContext, action_pool::ActionPool};
 use tcp_connection::error::TcpTargetError;
 
 use crate::{
-    actions::local_actions::register_set_upstream_vault_action,
-    connection::protocol::RemoteActionInvoke,
+    actions::local_actions::register_hello_world_action, connection::protocol::RemoteActionInvoke,
 };
 
 fn register_actions(pool: &mut ActionPool) {
     // Pool register here
-    register_set_upstream_vault_action(pool);
+    register_hello_world_action(pool);
 }
 
 pub fn client_action_pool() -> ActionPool {
@@ -19,13 +18,16 @@ pub fn client_action_pool() -> ActionPool {
     register_actions(&mut pool);
 
     // Add process events
-    pool.set_on_proc_begin(|ctx| Box::pin(on_proc_begin(ctx)));
+    pool.set_on_proc_begin(|ctx, args| Box::pin(on_proc_begin(ctx, args)));
 
     // Return
     pool
 }
 
-async fn on_proc_begin(ctx: &mut ActionContext) -> Result<(), TcpTargetError> {
+async fn on_proc_begin(
+    ctx: &ActionContext,
+    _args: &(dyn std::any::Any + Send + Sync),
+) -> Result<(), TcpTargetError> {
     // Is ctx remote
     let is_remote = ctx.is_remote();
 
@@ -34,7 +36,7 @@ async fn on_proc_begin(ctx: &mut ActionContext) -> Result<(), TcpTargetError> {
     let action_args_json = ctx.action_args_json().clone();
 
     // Get instance
-    let Some(instance) = ctx.instance_mut() else {
+    let Some(instance) = ctx.instance() else {
         return Err(TcpTargetError::Unsupported(
             "Missing ConnectionInstance in current context, this ActionPool does not support this call"
                 .to_string()));
@@ -49,7 +51,8 @@ async fn on_proc_begin(ctx: &mut ActionContext) -> Result<(), TcpTargetError> {
         };
 
         // Send
-        instance.write_msgpack(msg).await?;
+        let mut instance = instance.lock().await;
+        instance.write_msgpack(&msg).await?;
     }
 
     // Return OK, wait for client to execute Action locally