diff --git a/demo/progressive_random.py b/demo/progressive_random.py
index d2aba0f..50babe3 100644
--- a/demo/progressive_random.py
+++ b/demo/progressive_random.py
@@ -7,6 +7,7 @@
 n = 2
 choices = ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H']
 
+control.defaults.window_mode = True
 generators = [progressive_random.SequenceGenerator(choices, trials=trials_per_block, n=n) for b in range(0, num_of_blocks)]
 
 exp = design.Experiment("Progressive N-Back Task")
@@ -39,7 +40,9 @@
         stim = stimuli.TextLine(item, text_size=200)
         exp.clock.wait(1000 - fixation.present() - stim.preload())
         stim.present()
-        pressed_key, rt = exp.keyboard.wait(response_keys)
+
+        # Wait for response, remove duration param for forced-choice.
+        pressed_key, rt = exp.keyboard.wait(response_keys, duration=2000)
         exp.data.add([block.get_factor("block"), trial.get_factor("stimulus"), pressed_key, rt])
         exp.clock.wait(1000 - expyriment.stimuli.BlankScreen().present() - stim.unload())