Fixed the registeration issue