The move comes just days after President Donald Trump said Apple CEO Tim Cook promised him that the company would shift from Mexico to the U.S.