diff --git a/Project.toml b/Project.toml
index 875dd2e..442eff6 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "Walrus"
 uuid = "bec5164f-1c7a-4c32-8768-be9354254d6a"
 authors = ["Jago Stong-Wright <jagoosw@protonmail.com>"]
-version = "0.5.4"
+version = "0.5.5"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
diff --git a/src/wind_driven_stokes.jl b/src/wind_driven_stokes.jl
index c258127..1e5c4cf 100644
--- a/src/wind_driven_stokes.jl
+++ b/src/wind_driven_stokes.jl
@@ -34,7 +34,7 @@ module WindDrivenStokesParameterisation
 
 export WindDrivenStokesDrift, WindDrivenStokesDriftSetup
 
-using Roots
+using Roots, Adapt
 
 using Oceananigans.Architectures: on_architecture, CPU, architecture
 using Oceananigans.BuoyancyModels: g_Earth
@@ -169,4 +169,4 @@ function WindDrivenStokesDriftSetup(; wind, depth,
     return  UniformStokesDrift(; ∂z_uˢ, ∂z_vˢ, ∂t_uˢ, ∂t_vˢ, parameters = parameterisation)
 end
 
-end # module
\ No newline at end of file
+end # module