Make onDiagnoseFinished async