Hedge trimmer choke rod

I had to completely take apart this old HS 56 C to clean the spark arrestor. In doing so, I knocked loose the choke rod from the switch shaft it connects to.

Now the choke rod won’t stay in the switch shaft when I move the choke positions. How is this thing supposed to stay in there? Am I missing a piece? Did I break a piece of plastic? Nothing looks broken off.