Improve login test coverage