Fix test breaking due to prompt