diff --git a/crates/burn-wgpu/src/runtime.rs b/crates/burn-wgpu/src/runtime.rs
index 26d8d196b8..161a4802cc 100644
--- a/crates/burn-wgpu/src/runtime.rs
+++ b/crates/burn-wgpu/src/runtime.rs
@@ -79,6 +79,9 @@ pub struct RuntimeOptions {
 
 impl Default for RuntimeOptions {
     fn default() -> Self {
+        #[cfg(test)]
+        const DEFAULT_MAX_TASKS: usize = 1;
+        #[cfg(not(test))]
         const DEFAULT_MAX_TASKS: usize = 16;
 
         let tasks_max = match std::env::var("BURN_WGPU_MAX_TASKS") {