Fix the flaky test which depends on the unpredicable order of JSONObject keys