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.