diff --git a/cara/short_range_plots/model_scenarios.py b/cara/short_range_plots/model_scenarios.py
index b1444997..dbe0ddd7 100644
--- a/cara/short_range_plots/model_scenarios.py
+++ b/cara/short_range_plots/model_scenarios.py
@@ -152,7 +152,7 @@ def baseline_model(activity: str,
             ),
             infected=mc.InfectedPopulation(
                 number=1,
-                virus=virus_distributions['SARS_CoV_2'],
+                virus=virus_distributions['SARS_CoV_2_OMICRON'],
                 presence=models.SpecificInterval(((8.5, 12.5),)),
                 mask=exposure_mask,
                 activity=activity_distributions[activity],