comparison: use param_key API to store return_states