The U.S. should prioritize restricting China's AI capabilities over pursuing safety cooperation