Update for latest registration API