diff --git a/aepsych/generators/completion_criterion/min_total_outcome_occurrences.py b/aepsych/generators/completion_criterion/min_total_outcome_occurrences.py
index 3a27d68df..ea2ee0519 100644
--- a/aepsych/generators/completion_criterion/min_total_outcome_occurrences.py
+++ b/aepsych/generators/completion_criterion/min_total_outcome_occurrences.py
@@ -9,7 +9,7 @@
 
 from aepsych.config import Config, ConfigurableMixin
 
-from ax.modelbridge.completion_criterion import MinimumPreferenceOccurances
+from ax.modelbridge.transition_criterion import MinimumPreferenceOccurances
 
 
 class MinTotalOutcomeOccurrences(MinimumPreferenceOccurances, ConfigurableMixin):