Fix minor reg/login bug