Fix error on registration

- Fix error on registration.