Fix return url for login and register