diff --git a/tensorflow_privacy/privacy/analysis/dp_event.py b/tensorflow_privacy/privacy/analysis/dp_event.py
index ac45a64..b40b823 100644
--- a/tensorflow_privacy/privacy/analysis/dp_event.py
+++ b/tensorflow_privacy/privacy/analysis/dp_event.py
@@ -74,7 +74,7 @@ class SelfComposedDpEvent(DpEvent):
 @attr.s(frozen=True, slots=True, auto_attribs=True)
 class ComposedDpEvent(DpEvent):
   """A series of composed mechanisms."""
-  events: List[SelfComposedDpEvent]
+  events: List[DpEvent]
 
 
 @attr.s(frozen=True, slots=True, auto_attribs=True)