Fixed small issue when registering.