Fixed small oops in registration