Slight tweak to how the headless tests run.