diff --git a/src/AlgoimUtils/AlgoimUtils.jl b/src/AlgoimUtils/AlgoimUtils.jl
index 23a9d3f..73fcf49 100644
--- a/src/AlgoimUtils/AlgoimUtils.jl
+++ b/src/AlgoimUtils/AlgoimUtils.jl
@@ -532,7 +532,7 @@ end
 
 function compute_normal_displacement!(
     cache,
-    uₕ::FEFunction,
+    uₕ::CellField,
     cpₕ::FEFunction,
     phi::AlgoimCallLevelSetFunction,
     dt::Float64,