I'd say lock the wheel and mark it with something that won't rub off easy. Maybe check the marks right before you slide the coupler off.
I didn't mark mine and it took about 4 or 5 tries to get it straight
The tires looked straight, the wheel looked straight but when I drove it the steering wheel was crooked.
I got good at pulling the damned thing though
When I did my driveshaft bearing I marked it with spray paint. Maybe line up the "crack" in the coupler and hit it with a spray bomb? Thank God I marked THAT!