fix headless tests