Fixed tests so that they pass after making changes to register method