The following operations don't return actual information of the stages once the first stage is completed and the second stage is active:

stage = caseRuntimeDataService.getCaseInstanceStages(caseId, true, new QueryContext(0, 1));

List<CaseStage> stages = caseClient.getStages(CONTAINER_ID, caseId, true, 0, 1);