Fix Login/Register bug