policy_service protocol_state with smtpd_delay_reject

[postfix]

I notice when using SMTPD_DELAY_REJECT=yes and calling a 
CHECK_POLICY_SERVICE inside SMTPD_HELO_RESTRICTIONS it will report 
"protocol_state = RCPT", same as when you call the policy service from 
inside SMTPD_RECIPIENT_RESTRICTIONS.

Is there a way to pass a value from the CHECK_POLICY_SERVICE line (or 
any other method), to know which restriction section the policy service 
is being ran inside of?