diff --git a/ice_ocean_SIS2/Baltic/MOM_input b/ice_ocean_SIS2/Baltic/MOM_input
index 06860fbf45..daeb98a7c1 100644
--- a/ice_ocean_SIS2/Baltic/MOM_input
+++ b/ice_ocean_SIS2/Baltic/MOM_input
@@ -60,6 +60,8 @@ IO_LAYOUT = 1, 1                ! default = 0
 NK = 63                         !   [nondim]
                                 ! The number of model layers.
 
+DIAG_REMAP_Z_GRID_DEF = "FILE:OM3_zgrid.nc,zw"
+
 ! === module MOM_verticalGrid ===
 ! Parameters providing information about the vertical grid.
 
diff --git a/ice_ocean_SIS2/Baltic/MOM_parameter_doc.all b/ice_ocean_SIS2/Baltic/MOM_parameter_doc.all
index a3dfb1d71d..05493b7a40 100644
--- a/ice_ocean_SIS2/Baltic/MOM_parameter_doc.all
+++ b/ice_ocean_SIS2/Baltic/MOM_parameter_doc.all
@@ -455,7 +455,7 @@ APPLY_OBC_U = False             !   [Boolean] default = False
 APPLY_OBC_V = False             !   [Boolean] default = False
                                 ! If true, open boundary conditions may be set at some
                                 ! v-points, with the configuration controlled by OBC_CONFIG
-DIAG_REMAP_Z_GRID_DEF = ""      ! default = ""
+DIAG_REMAP_Z_GRID_DEF = "FILE:OM3_zgrid.nc,zw" ! default = ""
                                 ! This sets the file and variable names that define the
                                 ! vertical grid used for diagnostic output remapping to
                                 ! Z space. It should look like: