Both "stage" and "phase" are possible. In many cases the two words are interchangeable (in this sense), though "phase" suggests a larger undertaking over a longer time period; I wouldn't use "phase" for a minor sub-task of short duration. All your examples are correct (and would also be correct with "stage").
New words, one handy idiom, and a 2-minute quiz — delivered to your inbox to keep your streak alive.