diff --git a/sdk/monitor/azure-monitor-query/azure/monitor/query/_models.py b/sdk/monitor/azure-monitor-query/azure/monitor/query/_models.py
index 8cc132544836..3c66a5766b0f 100644
--- a/sdk/monitor/azure-monitor-query/azure/monitor/query/_models.py
+++ b/sdk/monitor/azure-monitor-query/azure/monitor/query/_models.py
@@ -11,7 +11,6 @@
 from ._generated.models import (
     Column as InternalColumn,
     QueryBody as InternalQueryBody,
-    BatchResponse as InternalBatchResponse,
     LogQueryRequest as InternalLogQueryRequest,
     MetricNamespace as InternalMetricNamespace,
     ErrorDetails as InternalErrorDetails
diff --git a/sdk/monitor/azure-monitor-query/azure/monitor/query/aio/_log_query_client_async.py b/sdk/monitor/azure-monitor-query/azure/monitor/query/aio/_log_query_client_async.py
index c0034ca36c20..5f41969ab489 100644
--- a/sdk/monitor/azure-monitor-query/azure/monitor/query/aio/_log_query_client_async.py
+++ b/sdk/monitor/azure-monitor-query/azure/monitor/query/aio/_log_query_client_async.py
@@ -7,8 +7,7 @@
 
 from typing import Any, Union, Sequence, Dict
 from azure.core.exceptions import HttpResponseError
-from azure.core.credentials import TokenCredential
-
+from azure.core.credentials_async import AsyncTokenCredential
 
 from .._generated.aio._monitor_query_client import MonitorQueryClient
 
@@ -21,12 +20,12 @@ class LogsClient(object):
     """LogsClient
 
     :param credential: The credential to authenticate the client
-    :type credential: ~azure.core.credentials.TokenCredential
+    :type credential: ~azure.core.credentials_async.AsyncTokenCredential
     :keyword endpoint: The endpoint to connect to. Defaults to 'https://api.loganalytics.io/v1'.
     :paramtype endpoint: str
     """
 
-    def __init__(self, credential: TokenCredential, **kwargs: Any) -> None:
+    def __init__(self, credential: AsyncTokenCredential, **kwargs: Any) -> None:
         self._endpoint = kwargs.pop('endpoint', 'https://api.loganalytics.io/v1')
         self._client = MonitorQueryClient(
             credential=credential,