Registration fix

Registration fix