diff --git a/core/storage/btree.rs b/core/storage/btree.rs
index 659dc2208..081c229ed 100644
--- a/core/storage/btree.rs
+++ b/core/storage/btree.rs
@@ -7798,8 +7798,7 @@ mod tests {
         },
         types::Text,
         vdbe::Register,
-        BufferPool, Completion, Connection, DatabaseStorage, IOContext, StepResult, WalFile,
-        WalFileShared,
+        BufferPool, Completion, Connection, IOContext, StepResult, WalFile, WalFileShared,
     };
     use std::{
         cell::RefCell,
@@ -9064,8 +9063,9 @@ mod tests {
         let io: Arc<dyn IO> = Arc::new(MemoryIO::new());
         let buffer_pool = BufferPool::begin_init(&io, page_size * 128);
 
-        let db_file =
-            DatabaseFile::new(io.open_file(":memory:", OpenFlags::Create, false).unwrap());
+        let db_file = Arc::new(DatabaseFile::new(
+            io.open_file(":memory:", OpenFlags::Create, false).unwrap(),
+        ));
 
         let wal_file = io.open_file("test.wal", OpenFlags::Create, false).unwrap();
         let wal_shared = WalFileShared::new_shared(wal_file).unwrap();
diff --git a/tests/integration/functions/test_wal_api.rs b/tests/integration/functions/test_wal_api.rs
index 20ec76e30..af1835a06 100644
--- a/tests/integration/functions/test_wal_api.rs
+++ b/tests/integration/functions/test_wal_api.rs
@@ -864,7 +864,7 @@ fn test_db_share_same_file() {
     let db_file = io
         .open_file(path.to_str().unwrap(), turso_core::OpenFlags::Create, false)
         .unwrap();
-    let db_file = turso_core::storage::database::DatabaseFile::new(db_file);
+    let db_file = Arc::new(turso_core::storage::database::DatabaseFile::new(db_file));
     let db1 = turso_core::Database::open_with_flags(
         io.clone(),
         path.to_str().unwrap(),