Fixed something in registration