diff --git a/aepsych/strategy.py b/aepsych/strategy.py
index c5348eebe..cb2aa163f 100644
--- a/aepsych/strategy.py
+++ b/aepsych/strategy.py
@@ -530,6 +530,7 @@ def get_config_options(cls, config: Config, name: Optional[str] = None) -> Dict:
         # Add an extra step at the end that we can `ask` endlessly.
         final_step = copy(step)
         final_step.completion_criteria = []
+        final_step._transition_criteria = []
         final_step.num_trials = -1
         steps.append(final_step)
 
diff --git a/tests/test_ax_integration.py b/tests/test_ax_integration.py
index a3ade8b4d..75b98ffc6 100644
--- a/tests/test_ax_integration.py
+++ b/tests/test_ax_integration.py
@@ -20,7 +20,6 @@
 from parameterized import parameterized_class
 
 
-@unittest.skip("Test is flaky. Skipping for now.")
 @parameterized_class(
     ("config_file", "should_ignore"),
     [