diff --git a/petab_select/plot.py b/petab_select/plot.py
index 06f3fb4..414927d 100644
--- a/petab_select/plot.py
+++ b/petab_select/plot.py
@@ -97,7 +97,7 @@ def upset(
     """
     # Get delta criterion values
     values = get_relative_criterion_values(
-        [model.get_criterion(Criterion.AICC) for model in models]
+        [model.get_criterion(criterion) for model in models]
     )
 
     # Sort by criterion value