Fix set session on register