diff --git a/src/main.ts b/src/main.ts
index faf326ecb..82712fc26 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -66,7 +66,7 @@ export async function shutdownAzureMonitor() {
 export async function flushAzureMonitor() {
     try {
         await (metrics.getMeterProvider() as MeterProvider).forceFlush();
-        await (trace.getTracerProvider() as BasicTracerProvider).forceFlush();
+        await (((trace.getTracerProvider() as ProxyTracerProvider).getDelegate()) as BasicTracerProvider).forceFlush();
         await (logs.getLoggerProvider() as LoggerProvider).forceFlush();
     } catch (err) {
         diag.error("Failed to flush telemetry", err);