The phrase 'maintain rwy heading' may have been withdrawn from official ATC use, but I believe that the TWR controller uses it to send an aircraft down the extended centerline. I correct for drift if when given that instuction.
I agree that 'after departure continue staight ahead' or some such is much better phraseology.