Armistead then served in Fort Towson, Oklahoma, Fort Washita near the Oklahoma border. Serving in the Mexican War, he was appointed brevet captain for Contreras and Churubusco, wounded at Chapultepec, and was appointed a brevet major for Molino del Rey and Chapultepec.