Update: I got it working.
Add these into the ajax_login()

if ( is_wp_error($user_signon) ){} else {
  wp_set_current_user($user_signon->ID);
  wp_set_auth_cookie($user_signon->ID);
}