diff --git a/veaiops/metrics/base.py b/veaiops/metrics/base.py
index b835bdb7..cc0960ed 100644
--- a/veaiops/metrics/base.py
+++ b/veaiops/metrics/base.py
@@ -17,10 +17,9 @@
 import functools
 from dataclasses import dataclass
 from datetime import datetime
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List, Literal, Optional
 
 from pydantic import BaseModel, ConfigDict, Field
-from typing_extensions import Literal
 
 from veaiops.metrics.timeseries import InputTimeSeries
 from veaiops.schema.documents import (
diff --git a/veaiops/metrics/timeseries.py b/veaiops/metrics/timeseries.py
index 44202acd..865ad8f5 100644
--- a/veaiops/metrics/timeseries.py
+++ b/veaiops/metrics/timeseries.py
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing_extensions import TypedDict
+from typing import TypedDict
 
 __all__ = ["InputTimeSeries"]
 
diff --git a/veaiops/utils/kb.py b/veaiops/utils/kb.py
index 66e0694d..9c224623 100644
--- a/veaiops/utils/kb.py
+++ b/veaiops/utils/kb.py
@@ -14,10 +14,9 @@
 
 import json
 from datetime import datetime, timezone
-from typing import List, Literal, Optional
+from typing import List, Literal, Optional, override
 
 from tenacity import retry, stop_after_attempt, wait_fixed
-from typing_extensions import override
 from volcengine.ApiInfo import ApiInfo
 from volcengine.viking_knowledgebase import Collection, Doc, Point, VikingKnowledgeBaseService