Clean up login button behavior