diff --git a/docs/examples/closed-loop-basic/workflow.md b/docs/examples/closed-loop-basic/workflow.md
index 45e093a1d..f612835f2 100644
--- a/docs/examples/closed-loop-basic/workflow.md
+++ b/docs/examples/closed-loop-basic/workflow.md
@@ -187,10 +187,10 @@ def runner_on_state(conditions):
         iv_1 = c['dots_left']
         iv_2 = c['dots_right']
         # get a timeline via sweetPea
-        timeline = trial_sequence(iv_1, iv_2, num_trials)[0]
+        timeline = trial_sequence(iv_1, iv_2, num_trials)
         print("Generated counterbalanced trial sequence.")
         # get js code via sweetBeaan
-        js_code = stimulus_sequence(timeline, iv_1, iv_2)
+        js_code = stimulus_sequence(timeline)
         print("Compiled experiment.")
         res.append(js_code)
     
@@ -452,10 +452,10 @@ We then define the function ``runner_on_state`` that uploads the experiment to F
         iv_1 = c['dots_left']
         iv_2 = c['dots_right']
         # get a timeline via sweetPea
-        timeline = trial_sequence(iv_1, iv_2, num_trials)[0]
+        timeline = trial_sequence(iv_1, iv_2, num_trials)
         print("Generated counterbalanced trial sequence.")
         # get js code via sweetBeaan
-        js_code = stimulus_sequence(timeline, iv_1, iv_2)
+        js_code = stimulus_sequence(timeline)
         print("Compiled experiment.")
         res.append(js_code)
 ```