U.S. President Donald Trump on Thursday said he told Apple CEO Tim Cook that he doesn’t want the tech giant to build its products in India, taking shots at the company’s moves to diversify production away from China and urging him to pivot Stateside.

“I had a little problem with Tim Cook yesterday,” Trump said. “I said to him, ‘my friend, I treated you very good. You’re coming here with $500 billion, but now I hear you’re building all over India.’ I don’t want you building in India.”

Trump was referencing Apple’s commitment of a $500 billion investment in the U.S. which was announced in February.

Apple has been ramping up production in India with the aim of making around 25% of global iPhones in the country in the next few years, as it looks to reduce reliance on China, where around 90% of its flagship smartphone is currently assembled.

“I said to Tim, I said, ‘Tim look, we treated you really good, we put up with all the plants that you build in China for years, now you got build us. We’re not interested in you building in India, India can take care of themselves … we want you to build here’,” Trump said.

The U.S. president added that Apple is going to be “upping” its production in the United States, without disclosing further details.

CNBC has reached out to Apple.

Trump made the comments about the U.S. tech giant while discussing Washington’s broader trade relations with India.

Trump said India is “one of the highest tariff nations in the world,” adding the country has offered a deal to the U.S. where “they are willing to literally charge us no tariff.”

Under the White House’s trade protectionist policies revealed in April, Trump has imposed a so-called “reciprocal tariff” of 26% on Indian goods, which has been temporarily lowered until July.

This developing story is being updated.