HydrostaticFreeSurfaceModel and TKEBudgetTerms

Hi all, that's more a question than an issue. So, I run a few simulations using the HydrostaticFreeSurfaceModel configuration and wanted to calculate some diagnostics for it's output so I found this package (awesome btw). I don't know if that's some silly question but I want to calculate some terms using the `TKEBudgetTerms`, for example the `BuoyancyProductionTerm `or the `AdvectionTerm `and it returns me an error when I use the `HydrostaticFreeSurfaceModel `config and I don't see why. Like, for the `AdvectionTerm `ADV = uᵢ∂ⱼ(uᵢuⱼ), why the `HydrostaticFreeSurfaceModel `not work? For the BP term I think that's related to the fact that the `w` in this configuration it's just a product of the divergence/convergence of `u` and `v`, but even though I thought that it may work. I don't know if that's something I'm missing and it's pretty obvious or not. Any help is appreciated! Cheers.