diff --git a/xoa_driver/internals/hli_v2/ports/port_l23/chimera/pe_custom_distribution.py b/xoa_driver/internals/hli_v2/ports/port_l23/chimera/pe_custom_distribution.py
index eca79edc..0c212fd8 100644
--- a/xoa_driver/internals/hli_v2/ports/port_l23/chimera/pe_custom_distribution.py
+++ b/xoa_driver/internals/hli_v2/ports/port_l23/chimera/pe_custom_distribution.py
@@ -56,7 +56,7 @@ async def delete(self) -> None:
         self.__observer.notify(observer.IndexEvents.DEL, self)
 
 
-class CustomDistributions(UserDict[int, "CustomDistribution"]):
+class CustomDistributions(UserDict):
     """Custom distributions"""
 
     def __init__(self, conn: "itf.IConnection", module_id: int, port_id: int) -> None: