Diversity is key to forests withstanding drought, research finds