diff --git a/threedigrid_builder/grid/boundary_conditions.py b/threedigrid_builder/grid/boundary_conditions.py
index 633fec0d..7113af51 100644
--- a/threedigrid_builder/grid/boundary_conditions.py
+++ b/threedigrid_builder/grid/boundary_conditions.py
@@ -18,7 +18,11 @@
 
 
 GROUNDWATER_BOUNDARY_TYPES = frozenset(
-    {BoundaryType.GROUNDWATERLEVEL, BoundaryType.GROUNDWATERDISCHARGE}
+    {
+        BoundaryType.GROUNDWATERLEVEL,
+        BoundaryType.GROUNDWATERDISCHARGE,
+        BoundaryType.GROUNDWATER_TOTAL_DISCHARGE_2D,
+    }
 )
 
 
@@ -270,6 +274,7 @@ def boundary_type_is_groundwater(type: BoundaryType) -> bool:
         return type in {
             BoundaryType.GROUNDWATERLEVEL,
             BoundaryType.GROUNDWATERDISCHARGE,
+            BoundaryType.GROUNDWATER_TOTAL_DISCHARGE_2D,
         }
 
     def check_edge_coord(