Update tests to call new theatre setup.