diff --git a/src/dynamics/diagnostic_variables.jl b/src/dynamics/diagnostic_variables.jl
index 70dc077b2..9fa751e4f 100644
--- a/src/dynamics/diagnostic_variables.jl
+++ b/src/dynamics/diagnostic_variables.jl
@@ -400,6 +400,9 @@ struct DiagnosticVariables{
     scale::Base.RefValue{NF}
 end
 
+DiagnosticVariables(SG::SpectralGrid, model::Barotropic) = DiagnosticVariables(SG)
+DiagnosticVariables(SG::SpectralGrid, model::ShallowWater) = DiagnosticVariables(SG)
+
 # decide on spectral resolution `nbands` of radiation schemes
 function DiagnosticVariables(SG::SpectralGrid, model::PrimitiveEquation)
     ntracers = length(model.tracers)
diff --git a/src/models/primitive_dry.jl b/src/models/primitive_dry.jl
index 07fa2c781..45f3e521f 100644
--- a/src/models/primitive_dry.jl
+++ b/src/models/primitive_dry.jl
@@ -85,7 +85,7 @@ $(TYPEDFIELDS)"""
     optical_depth::OD = ZeroOpticalDepth(spectral_grid)
     shortwave_radiation::SW = NoShortwave(spectral_grid)
     longwave_radiation::LW = JeevanjeeRadiation(spectral_grid)
-    stochastic_physics::SP = StochasticallyPerturbedParameterizationTendencies(spectral_grid)
+    stochastic_physics::SP = NoStochasticPhysics(spectral_grid)
     
     # NUMERICS
     time_stepping::TS = Leapfrog(spectral_grid)
diff --git a/src/models/primitive_wet.jl b/src/models/primitive_wet.jl
index c22a14378..eeaceaf38 100644
--- a/src/models/primitive_wet.jl
+++ b/src/models/primitive_wet.jl
@@ -96,7 +96,7 @@ $(TYPEDFIELDS)"""
     optical_depth::OD = ZeroOpticalDepth(spectral_grid)
     shortwave_radiation::SW = NoShortwave(spectral_grid)
     longwave_radiation::LW = JeevanjeeRadiation(spectral_grid)
-    stochastic_physics::SP = StochasticallyPerturbedParameterizationTendencies(spectral_grid)
+    stochastic_physics::SP = NoStochasticPhysics(spectral_grid)
     
     # NUMERICS
     time_stepping::TS = Leapfrog(spectral_grid)