diff --git a/src/Drivers/Drivers.jl b/src/Drivers/Drivers.jl
index 733695e515359d40b16a3aa6c218bf1caa21ffb1..3e85fe2d3bcb80b1d629f8e33ec96a4e0a79b3fb 100644
--- a/src/Drivers/Drivers.jl
+++ b/src/Drivers/Drivers.jl
@@ -16,6 +16,7 @@ using ComponentArrays
 using Dates
 using DimensionalData
 using Flatten
+using IfElse
 using ModelParameters
 using LinearAlgebra
 using Reexport
diff --git a/src/Drivers/courant_step.jl b/src/Drivers/courant_step.jl
index f9e609a7edc46bd1a1634c6bcc30a6ee2733ead6..70fd96850293cae5b9135b3fdbdddf28c6b21993 100644
--- a/src/Drivers/courant_step.jl
+++ b/src/Drivers/courant_step.jl
@@ -24,6 +24,6 @@ function (cfl::CFLStepLimiter{<:Tile,Nothing})(u,p,t)
     end
 end
 function CFLStepLimiter(tile::HeatOnlyTile; courant_number=1//2, default_dt=60.0, iip=true)
-    cfl = iip ? CFLStepLimiter(tile, zero(dustrip(Δ(setup.grid))), default_dt) : CFLHeatState(tile, nothing, default_dt)
+    cfl = iip ? CFLStepLimiter(tile, zero(dustrip(Δ(tile.grid))), default_dt) : CFLStepLimiter(tile, nothing, default_dt)
     StepsizeLimiter(cfl; safety_factor=courant_number, max_step=true)
 end