Bug fixes for  new auth register