diff --git a/crates/cuddle-ci/src/rust_service/ssh_agent.rs b/crates/cuddle-ci/src/rust_service/ssh_agent.rs
index 5c6dea1..4d66b1c 100644
--- a/crates/cuddle-ci/src/rust_service/ssh_agent.rs
+++ b/crates/cuddle-ci/src/rust_service/ssh_agent.rs
@@ -27,13 +27,13 @@ impl DaggerMiddleware for SshAgent {
 }
 
 pub trait SshAgentExt {
-    fn with_socket(&mut self, socket: dagger_sdk::Socket) -> &mut Self {
+    fn with_ssh_agent(&mut self, socket: dagger_sdk::Socket) -> &mut Self {
         self
     }
 }
 
 impl SshAgentExt for RustService {
-    fn with_socket(&mut self, socket: dagger_sdk::Socket) -> &mut Self {
+    fn with_ssh_agent(&mut self, socket: dagger_sdk::Socket) -> &mut Self {
         self.with_stage(super::RustServiceStage::BeforeDeps(Arc::new(
             SshAgent::new(socket.clone()),
         )))
@@ -44,7 +44,7 @@ impl SshAgentExt for RustService {
 }
 
 impl SshAgentExt for LeptosService {
-    fn with_socket(&mut self, socket: dagger_sdk::Socket) -> &mut Self {
+    fn with_ssh_agent(&mut self, socket: dagger_sdk::Socket) -> &mut Self {
         self.with_stage(super::RustServiceStage::BeforeDeps(Arc::new(
             SshAgent::new(socket.clone()),
         )))