diff --git a/examples/tree_1d_dgsem/elixir_euler_blast_wave_neuralnetwork_perssonperaire.jl b/examples/tree_1d_dgsem/elixir_euler_blast_wave_neuralnetwork_perssonperaire.jl
index 449dd17..1fdfd2b 100644
--- a/examples/tree_1d_dgsem/elixir_euler_blast_wave_neuralnetwork_perssonperaire.jl
+++ b/examples/tree_1d_dgsem/elixir_euler_blast_wave_neuralnetwork_perssonperaire.jl
@@ -10,6 +10,9 @@ using OrdinaryDiffEq
 using TrixiSmartShockFinder
 using Trixi
 
+NeuralNetworkPerssonPeraire = TrixiSmartShockFinder.NeuralNetworkPerssonPeraire
+NeuralNetworkRayHesthaven = TrixiSmartShockFinder.NeuralNetworkRayHesthaven
+
 # This elixir was one of the setups used in the following master thesis:
 # - Julia Odenthal (2021)
 #   Shock capturing with artificial neural networks
diff --git a/examples/tree_1d_dgsem/elixir_euler_blast_wave_neuralnetwork_rayhesthaven.jl b/examples/tree_1d_dgsem/elixir_euler_blast_wave_neuralnetwork_rayhesthaven.jl
index 5105efb..2ac20ae 100644
--- a/examples/tree_1d_dgsem/elixir_euler_blast_wave_neuralnetwork_rayhesthaven.jl
+++ b/examples/tree_1d_dgsem/elixir_euler_blast_wave_neuralnetwork_rayhesthaven.jl
@@ -10,6 +10,9 @@ using OrdinaryDiffEq
 using TrixiSmartShockFinder
 using Trixi
 
+NeuralNetworkPerssonPeraire = TrixiSmartShockFinder.NeuralNetworkPerssonPeraire
+NeuralNetworkRayHesthaven = TrixiSmartShockFinder.NeuralNetworkRayHesthaven
+
 # This elixir was one of the setups used in the following master thesis:
 # - Julia Odenthal (2021)
 #   Shock capturing with artificial neural networks