diff --git a/tig-runtime/src/main.rs b/tig-runtime/src/main.rs
index 2321236..100ce14 100644
--- a/tig-runtime/src/main.rs
+++ b/tig-runtime/src/main.rs
@@ -174,9 +174,8 @@ pub fn compute_solution(
                 let ptx = Ptx::from_src(modified_ptx);
                 let ctx = CudaContext::new(gpu_device)?;
                 ctx.set_blocking_synchronize()?;
-                ctx.enable_automatic_fuel_check();
                 let module = ctx.load_module(ptx)?;
-                let stream = ctx.default_stream();
+                let stream = ctx.fuel_check_stream();
                 let prop = get_device_prop(gpu_device as i32)?;
 
                 let challenge = $c::Challenge::generate_instance(