Although Dark Phoenix is a relatively more grounded adaptation of The Dark Phoenix Saga, this version definitely has more of a cosmic feel, namely because Jean’s transformation into Phoenix occurs during a rescue mission in space and Jessica Chastain’s still-unidentified character is an alien shapeshifter.
So it’s peculiar that rather than keep that cosmic flavor going in the climax, it was decided ahead of reshoots to keep the action Earthbound instead.