diff --git a/scripts/py_matter_idl/matter_idl/generators/idl/__init__.py b/scripts/py_matter_idl/matter_idl/generators/idl/__init__.py
index 15cf8c5113b405..f53e35a7aba8e0 100644
--- a/scripts/py_matter_idl/matter_idl/generators/idl/__init__.py
+++ b/scripts/py_matter_idl/matter_idl/generators/idl/__init__.py
@@ -13,12 +13,12 @@
 # limitations under the License.
 
 import os
-from typing import List, Union
+from typing import Union
 
 from matter_idl.generators import CodeGenerator, GeneratorStorage
-from matter_idl.matter_idl_types import (AccessPrivilege, ApiMaturity, Attribute, AttributeQuality, AttributeStorage, Cluster,
-                                         ClusterSide, Command, CommandQuality, Event, EventPriority, EventQuality, FieldQuality,
-                                         Idl, Struct, StructQuality, StructTag)
+from matter_idl.matter_idl_types import (AccessPrivilege, ApiMaturity, Attribute, AttributeQuality, AttributeStorage, ClusterSide,
+                                         Command, CommandQuality, Event, EventPriority, EventQuality, FieldQuality, Idl,
+                                         StructQuality, StructTag)
 
 
 def human_text_string(value: Union[ClusterSide, StructTag, StructQuality, EventPriority, EventQuality, AccessPrivilege, AttributeQuality, CommandQuality, ApiMaturity, AttributeStorage]) -> str:
diff --git a/scripts/py_matter_idl/matter_idl/test_idl_generator.py b/scripts/py_matter_idl/matter_idl/test_idl_generator.py
index cc8e26a56bac62..ea1347fb0e7d4c 100755
--- a/scripts/py_matter_idl/matter_idl/test_idl_generator.py
+++ b/scripts/py_matter_idl/matter_idl/test_idl_generator.py
@@ -17,10 +17,7 @@
 import os
 import sys
 import unittest
-from dataclasses import dataclass, field
-from typing import List
-
-import yaml
+from typing import Optional
 
 try:
     from matter_idl.matter_idl_parser import CreateParser
diff --git a/scripts/py_matter_idl/matter_idl/xml_parser.py b/scripts/py_matter_idl/matter_idl/xml_parser.py
index 6fcd987d78661f..d20bdb9dda67ec 100755
--- a/scripts/py_matter_idl/matter_idl/xml_parser.py
+++ b/scripts/py_matter_idl/matter_idl/xml_parser.py
@@ -16,6 +16,7 @@
 
 import logging
 import os
+from typing import Optional
 
 try:
     from matter_idl.zapxml import ParseSource, ParseXmls