diff --git a/src/lorenzpy/simulations/autonomous_flows.py b/src/lorenzpy/simulations/autonomous_flows.py
index 133d3a8..04fbfa1 100644
--- a/src/lorenzpy/simulations/autonomous_flows.py
+++ b/src/lorenzpy/simulations/autonomous_flows.py
@@ -1,4 +1,6 @@
 """Autonomous flows."""
+from __future__ import annotations
+
 from typing import Callable
 
 import numpy as np
diff --git a/src/lorenzpy/simulations/discrete_maps.py b/src/lorenzpy/simulations/discrete_maps.py
index 3b6146b..996a88a 100644
--- a/src/lorenzpy/simulations/discrete_maps.py
+++ b/src/lorenzpy/simulations/discrete_maps.py
@@ -1,4 +1,6 @@
 """Discrete maps."""
+from __future__ import annotations
+
 import numpy as np
 
 from .base import _BaseSimIterate
diff --git a/src/lorenzpy/simulations/driven_systems.py b/src/lorenzpy/simulations/driven_systems.py
index f29c596..c9831b9 100644
--- a/src/lorenzpy/simulations/driven_systems.py
+++ b/src/lorenzpy/simulations/driven_systems.py
@@ -1,4 +1,6 @@
 """Driven Systems."""
+from __future__ import annotations
+
 import numpy as np
 
 from .base import _BaseSimFlowDriven