I haven’t the answer to your question, but I do suggest (if you haven’t already) you explore the potential of midi retrospective record. This is recording your input even when you are not recording, providing the track is record enabled.
Otherwise set punch in a couple of beats early?