After taking a shower this evening, I pushed in the shower diverter stem. First the top part (the part you pull up/push down on) came off. When I tried to screw it back on, a square plastic piece fell out and the whole stem came out. Can someone give me an idea of how this is supposed to fit back into the faucet?

Thanks in advance.