1 files changed, 3 insertions, 4 deletions
diff --git a/crates/daemon/src/aclient/database/mod.rs b/crates/daemon/src/aclient/database/mod.rs
index 36049f80..5e25a0e9 100644
--- a/crates/daemon/src/aclient/database/mod.rs
+++ b/crates/daemon/src/aclient/database/mod.rs
@@ -51,10 +51,9 @@ pub(crate) struct OptFilters {
     pub(crate) include_duplicates: bool,
 }
 
-pub(crate) async fn current_context() -> eyre::Result<Context> {
-    let session = env::var("ATUIN_SESSION").map_err(|_| {
-        eyre::eyre!("Failed to find $ATUIN_SESSION in the environment. Check that you have correctly set up your shell.")
-    })?;
+pub(crate) async fn current_context(session: String) -> eyre::Result<Context> {
+    // TODO(@bpeetz): More of this needs to be moved to the client <2026-07-20>
+
     let hostname = get_host_user();
     let cwd = utils::get_current_dir();
     let host_id = Settings::host_id().await?;