You probably already resolved this but I wanted to mention, if you twisted the headset off, a block and hammer is probably all you need to put it back. Start with little raps until you see it going in straight and then gently hammer the rest of thw way. I do this on my older bikes where the headsets seem to fit looser.
__________________