I can lock mine in any position, don't have to pull it right out. I can only just pull it out far enough for the shaft to clear the socket. When it cleared I gave it a full turn and reinserted into the socket. It seemed to give it just enough "twisting spring" to help it jam in the socket when released. Mine has a "D" profile shaft, I think they all do.
edit; I turned it one full turn and reinserted it ..