diff --git a/burn-dataset/Cargo.toml b/burn-dataset/Cargo.toml
index cd38cf562..47b2837c3 100644
--- a/burn-dataset/Cargo.toml
+++ b/burn-dataset/Cargo.toml
@@ -23,14 +23,14 @@ sqlite = ["__sqlite-shared", "dep:rusqlite"]
 sqlite-bundled = ["__sqlite-shared", "rusqlite/bundled"]
 
 # internal
-__sqlite-shared = ["dep:r2d2", "dep:r2d2_sqlite", "dep:serde_rusqlite", "dep:image"]
+__sqlite-shared = ["dep:r2d2", "dep:r2d2_sqlite", "dep:serde_rusqlite", "dep:image", "dep:gix-tempfile"]
 
 [dependencies]
 csv = {workspace = true}
 derive-new = {workspace = true}
 dirs = {workspace = true}
 fake = {workspace = true, optional = true}
-gix-tempfile = {workspace = true}
+gix-tempfile = {workspace = true, optional = true}
 hound = {version = "3.5.0", optional = true}
 image = {version = "0.24.6", features = ["png"], optional = true}
 r2d2 = {workspace = true, optional = true}