Update for new login API