Add waitForDiagnosticsSource