diff --git a/cdp_use/generator/command_generator.py b/cdp_use/generator/command_generator.py
index 5dde7c8..efd4ddf 100644
--- a/cdp_use/generator/command_generator.py
+++ b/cdp_use/generator/command_generator.py
@@ -26,8 +26,7 @@ def generate_commands(self, domain: Dict[str, Any]) -> str:
         commands = domain.get("commands", [])
 
         # Always add basic imports
-        self.imports.add("from typing import Any, Dict, List, Optional, Union")
-        self.imports.add("from typing_extensions import NotRequired, TypedDict")
+        self.imports.add("from typing import Any, Dict, List, NotRequired, Optional, TypedDict, Union")
 
         # Start with auto-generated header
         content = """# This file is auto-generated by the CDP protocol generator.
diff --git a/cdp_use/generator/event_generator.py b/cdp_use/generator/event_generator.py
index d0de3e7..cdb90c3 100644
--- a/cdp_use/generator/event_generator.py
+++ b/cdp_use/generator/event_generator.py
@@ -26,8 +26,7 @@ def generate_events(self, domain: Dict[str, Any]) -> str:
         events = domain.get("events", [])
 
         # Always add basic imports
-        self.imports.add("from typing import Any, Dict, List, Optional, Union")
-        self.imports.add("from typing_extensions import NotRequired, TypedDict")
+        self.imports.add("from typing import Any, Dict, List, NotRequired, Optional, TypedDict, Union")
 
         # Start with auto-generated header
         content = """# This file is auto-generated by the CDP protocol generator.
diff --git a/cdp_use/generator/type_generator.py b/cdp_use/generator/type_generator.py
index 6e8273a..7adbf34 100644
--- a/cdp_use/generator/type_generator.py
+++ b/cdp_use/generator/type_generator.py
@@ -26,8 +26,7 @@ def generate_types(self, domain: Dict[str, Any]) -> str:
         types = domain.get("types", [])
 
         # Always add basic imports
-        self.imports.add("from typing import Any, Dict, List, Optional, Union")
-        self.imports.add("from typing_extensions import NotRequired, TypedDict")
+        self.imports.add("from typing import Any, Dict, List, NotRequired, Optional, TypedDict, Union")
 
         # First pass: collect all type names that will be defined in this domain
         for type_def in types:
@@ -109,7 +108,7 @@ def generate_literal_type(
         enum_values = type_def["enum"]
 
         # Add Literal import
-        self.imports.add("from typing_extensions import Literal")
+        self.imports.add("from typing import Literal")
 
         content = ""