Fix --headless