Add waits to login e2e to fix CI errors and improve the reliability