diff --git a/lib/src/deploy.rs b/lib/src/deploy.rs
index 6f9e7d306..e2b065f72 100644
--- a/lib/src/deploy.rs
+++ b/lib/src/deploy.rs
@@ -268,7 +268,7 @@ pub(crate) async fn pull(
     Ok(Box::new((*import).into()))
 }
 
-pub(crate) async fn cleanup(sysroot: &SysrootLock) -> Result<()> {
+pub(crate) async fn cleanup(sysroot: &Storage) -> Result<()> {
     // We create clones (just atomic reference bumps) here to move to the thread.
     let repo = sysroot.repo();
     let sysroot = sysroot.sysroot.clone();
@@ -329,7 +329,7 @@ pub(crate) fn get_base_commit(repo: &ostree::Repo, commit: &str) -> Result<Optio
 
 #[context("Writing deployment")]
 async fn deploy(
-    sysroot: &SysrootLock,
+    sysroot: &Storage,
     merge_deployment: Option<&Deployment>,
     stateroot: &str,
     image: &ImageState,
@@ -383,7 +383,7 @@ fn origin_from_imageref(imgref: &ImageReference) -> Result<glib::KeyFile> {
 /// Stage (queue deployment of) a fetched container image.
 #[context("Staging")]
 pub(crate) async fn stage(
-    sysroot: &SysrootLock,
+    sysroot: &Storage,
     stateroot: &str,
     image: &ImageState,
     spec: &RequiredHostSpec<'_>,
diff --git a/lib/src/kargs.rs b/lib/src/kargs.rs
index 90119ab34..6f6bdec45 100644
--- a/lib/src/kargs.rs
+++ b/lib/src/kargs.rs
@@ -10,10 +10,10 @@ use ostree_ext::ostree::Deployment;
 use ostree_ext::prelude::Cast;
 use ostree_ext::prelude::FileEnumeratorExt;
 use ostree_ext::prelude::FileExt;
-use ostree_ext::sysroot::SysrootLock;
 use serde::Deserialize;
 
 use crate::deploy::ImageState;
+use crate::store::Storage;
 
 /// The kargs.d configuration file.
 #[derive(Deserialize)]
@@ -96,7 +96,7 @@ fn get_kargs_from_ostree(
 /// karg, but applies the diff between the bootc karg files in /usr/lib/bootc/kargs.d
 /// between the booted deployment and the new one.
 pub(crate) fn get_kargs(
-    sysroot: &SysrootLock,
+    sysroot: &Storage,
     merge_deployment: &Deployment,
     fetched: &ImageState,
 ) -> Result<Vec<String>> {