diff --git a/docs/make.jl b/docs/make.jl
index 168ef896761d830dccb1b6c12c482579f2f06d95..bfe244190fa126154a11cc75223a8fd3bf06a8b4 100644
--- a/docs/make.jl
+++ b/docs/make.jl
@@ -10,6 +10,8 @@ const modules = [
        CryoGrid.Physics,
        CryoGrid.Boundaries,
        CryoGrid.HeatConduction,
+       CryoGrid.WaterBalance,
+       CryoGrid.Soils,
        CryoGrid.SEB,
        CryoGrid.Strat,
        CryoGrid.Presets,
@@ -33,6 +35,8 @@ makedocs(modules=modules,
                        "Utilities" => "api/utils.md",
                        "Physics" => [
                             "Heat Conduction" => "api/heat_conduction.md",
+                            "Water Balance" => "api/water_balance.md",
+                            "Soils" => "api/soils.md",
                             "Surface Energy Balance" => "api/seb.md",
                        ],
                        "Stratigraphy" => "api/strat.md",
diff --git a/docs/src/api/soils.md b/docs/src/api/soils.md
new file mode 100644
index 0000000000000000000000000000000000000000..ca662091837210b4fa5c214f4eb65fe70386b832
--- /dev/null
+++ b/docs/src/api/soils.md
@@ -0,0 +1,7 @@
+# Soils
+
+```@autodocs
+Modules = [Soils]
+Private = false
+Order = [:type, :function, :macro]
+```
\ No newline at end of file
diff --git a/docs/src/api/water_balance.md b/docs/src/api/water_balance.md
new file mode 100644
index 0000000000000000000000000000000000000000..b4d5bef886a59ee25cc48a81acd1d5f75c42ac42
--- /dev/null
+++ b/docs/src/api/water_balance.md
@@ -0,0 +1,7 @@
+# Water balance
+
+```@autodocs
+Modules = [WaterBalance]
+Private = false
+Order = [:type, :function, :macro]
+```
\ No newline at end of file
diff --git a/src/Physics/Soils/Soils.jl b/src/Physics/Soils/Soils.jl
index 7df337f7ca046e7697f0c0dc512afdb824862fe5..1eee115380ee5bfccad06f74597a6357a2fadc72 100644
--- a/src/Physics/Soils/Soils.jl
+++ b/src/Physics/Soils/Soils.jl
@@ -7,7 +7,7 @@ import CryoGrid.Physics.HeatConduction: Enthalpy, Temperature, totalwater, liqui
 using CryoGrid.Numerics
 using CryoGrid.Numerics: heaviside
 using CryoGrid.Physics.HeatConduction
-using CryoGrid.Physics.WaterBalance: VanGenuchten
+using CryoGrid.Physics.WaterBalance
 using CryoGrid.Utils
 
 using Base: @propagate_inbounds