Minor bug in extracting control_flow.record_deterministic_values