Small cleanup in login