Species diversification to help forests adapt to extreme drought