Fix headless build