diff --git a/src/ert/gui/plottery/plot_config_factory.py b/src/ert/gui/plottery/plot_config_factory.py
index fa015bb3972..34d988646c6 100644
--- a/src/ert/gui/plottery/plot_config_factory.py
+++ b/src/ert/gui/plottery/plot_config_factory.py
@@ -1,5 +1,11 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
 from ert.gui.plottery import PlotConfig
-from ert.gui.tools.plot.plot_api import PlotApiKeyDefinition
+
+if TYPE_CHECKING:
+    from ert.gui.tools.plot.plot_api import PlotApiKeyDefinition
 
 
 class PlotConfigFactory:
diff --git a/src/ert/gui/plottery/plot_context.py b/src/ert/gui/plottery/plot_context.py
index 185b9f96a87..e92a2f47918 100644
--- a/src/ert/gui/plottery/plot_context.py
+++ b/src/ert/gui/plottery/plot_context.py
@@ -2,12 +2,11 @@
 
 from typing import TYPE_CHECKING, ClassVar, List, Optional
 
-from ert.gui.tools.plot.plot_api import EnsembleObject
-
 if TYPE_CHECKING:
     from pandas import DataFrame
 
     from ert.gui.plottery import PlotConfig
+    from ert.gui.tools.plot.plot_api import EnsembleObject
 
 
 class PlotContext:
diff --git a/src/ert/gui/plottery/plots/ensemble.py b/src/ert/gui/plottery/plots/ensemble.py
index 2171f03c04b..d14786b9d8c 100644
--- a/src/ert/gui/plottery/plots/ensemble.py
+++ b/src/ert/gui/plottery/plots/ensemble.py
@@ -6,7 +6,6 @@
 import pandas as pd
 
 from ert.gui.plottery.plots.history import plotHistory
-from ert.gui.tools.plot.plot_api import EnsembleObject
 from ert.summary_key_type import is_rate
 
 from .observations import plotObservations
@@ -18,6 +17,7 @@
     from matplotlib.figure import Figure
 
     from ert.gui.plottery import PlotConfig, PlotContext
+    from ert.gui.tools.plot.plot_api import EnsembleObject
 
 
 class EnsemblePlot: