diff --git a/examples/elixir_euler_potential_temperature_dry_bubble.jl b/examples/elixir_euler_potential_temperature_dry_bubble.jl
index 1491ee8..33216b2 100644
--- a/examples/elixir_euler_potential_temperature_dry_bubble.jl
+++ b/examples/elixir_euler_potential_temperature_dry_bubble.jl
@@ -1,6 +1,6 @@
 using OrdinaryDiffEq
-using Trixi
-using TrixiAtmo
+using Trixi, TrixiAtmo
+using TrixiAtmo: source_terms_gravity
 
 function initial_condition_warm_bubble(x, t,
                                        equations::CompressibleEulerPotentialTemperatureEquations2D)
diff --git a/examples/elixir_euler_potential_temperature_gravity_wave.jl b/examples/elixir_euler_potential_temperature_gravity_wave.jl
index 64900ef..880e41c 100644
--- a/examples/elixir_euler_potential_temperature_gravity_wave.jl
+++ b/examples/elixir_euler_potential_temperature_gravity_wave.jl
@@ -1,6 +1,6 @@
 using OrdinaryDiffEq
-using Trixi
-using TrixiAtmo
+using Trixi, TrixiAtmo
+using TrixiAtmo: source_terms_gravity
 
 function initial_condition_gravity_wave(x, t,
                                         equations::CompressibleEulerPotentialTemperatureEquations2D)