diff --git a/crates/cuddle-ci/src/rust_service/sqlx.rs b/crates/cuddle-ci/src/rust_service/sqlx.rs
index 104613b..9bd4e7d 100644
--- a/crates/cuddle-ci/src/rust_service/sqlx.rs
+++ b/crates/cuddle-ci/src/rust_service/sqlx.rs
@@ -3,6 +3,8 @@ use dagger_sdk::Container;
 
 use crate::dagger_middleware::DaggerMiddleware;
 
+use super::RustService;
+
 pub struct Sqlx {}
 #[async_trait]
 impl DaggerMiddleware for Sqlx {
@@ -10,3 +12,17 @@ impl DaggerMiddleware for Sqlx {
         Ok(container.with_env_variable("SQLX_OFFLINE", "true"))
     }
 }
+
+pub trait SqlxExt {
+    fn with_sqlx(&mut self) -> &mut Self {
+        self
+    }
+}
+
+impl SqlxExt for RustService {
+    fn with_sqlx(&mut self) -> &mut Self {
+        self.with_stage(super::RustServiceStage::BeforeBuild(Box::new(Sqlx {})));
+
+        self
+    }
+}